tensorlfow 复制与拼接

来源:互联网 发布:算法精解源码 编辑:程序博客网 时间:2024/06/06 00:53
import tensorflow as tfsess = tf.Session()feature = tf.ones([3,4,2,5], tf.int32)print 'feature:\n', sess.run(feature),"\n"nums_origin = tf.constant([1,2,3])print 'nums_origin:\n', sess.run(nums_origin),"\n"nums = tf.expand_dims(nums_origin,-1)print 'nums: 3x1\n', sess.run(nums),"\n"nums = tf.tile(nums, [1,tf.shape(feature)[1]])print 'nums: 3x4\n', sess.run(nums),"\n"nums = tf.expand_dims(nums,-1)print 'nums: 3x4x1\n', sess.run(nums),"\n"nums = tf.tile(nums, [1,1,tf.shape(feature)[2]])print 'nums: 3x4x2\n', sess.run(nums),"\n"nums = tf.expand_dims(nums,-1)print 'nums: 3x4x2x1\n', sess.run(nums),"\n"con = tf.concat([feature, nums], 3) print 'concat:\n', sess.run(con),"\n"#nums  = tf.reshape(nums,[tf.shape(feature)[0], tf.shape(feature)[1], tf.shape(feature)[2], 1])
原创粉丝点击