Как реплицировать столбец в тензорном потоке?

Question

У меня есть тензор с формой (k, 1), и я хочу повторить первый столбец n раз.То есть результирующий тензор должен иметь форму (k, n) со значениями матрицы, скопированными из первого столбца.Как этого добиться с помощью tf.tile?

Answer 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.]]