我应该使用哪个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]])