如何在张量流中复制列?

时间:2018-10-01 16:05:53

标签: tensorflow tensor

我有一个形状为(k,1)的张量,我想复制第一列n次。也就是说,所得张量应具有形状(k,n),并具有从第一列复制的矩阵值。如何使用tf.tile实现这一目标?

1 个答案:

答案 0 :(得分:1)

您可以使用tf.tile()操作。例如:

a = tf.constant([[1], [2], [3]], dtype = tf.float32)
c = tf.tile(a, [1,3])
with tf.Session() as sess:
    print(sess.run(c))

返回:

[[1. 1. 1.]
 [2. 2. 2.]
 [3. 3. 3.]]