tensorflow - 使用tf.tile复制行张量？

Question

我有一个张量，它只是一个向量，vector = [0.5 0.4]和tf.shape表示它具有shape =（1，），我想将向量复制m次并具有[m，2]的形状，所以对于m = 2，matrix = [[0.5 0.4], [0.5 0.4]]。如何使用tf.tile实现该目标？

Answer 1

取以下内容，vec是向量，multiply是您的m，即重复向量的次数。在向量上执行tf.tile，然后使用tf.reshape将其重新整形为所需的结构。

import tensorflow as tf

vec = tf.constant([1, 2, 3, 4])
multiply = tf.constant([3])

matrix = tf.reshape(tf.tile(vec, multiply), [ multiply[0], tf.shape(vec)[0]])
with tf.Session() as sess:
    print(sess.run([matrix]))

结果是：

[array([[1, 2, 3, 4],
       [1, 2, 3, 4],
       [1, 2, 3, 4]], dtype=int32)]