我应该使用哪个tensorflow API来制作这个张量？

Question

假设存在张量A=[[1, 2], [2, 3], [4,5]]

我想基于A

创建新的张量

A' = [[1, 2], [1, 2], [1, 2], [2, 3], [2, 3], [2, 3], [4, 5], [4, 5], [4, 5]]

哪个tensorflow API可以帮助我做到这一点？

Answer 1

Tensorflow具有平铺方法，用于在某个轴上重复张量的值。该方法接受张量和列表，该张量和列表表示每个轴的重复次数。

A = tf.constant([[1, 2], [2, 3], [4,5]])
A_ = tf.reshape(tf.tile(A, [1,3]),[-1,2])

A_.eval(session=session)
#array([[1, 2],
       [1, 2],
       [1, 2],
       [2, 3],
       [2, 3],
       [2, 3],
       [4, 5],
       [4, 5],
       [4, 5]])