从维度为 [N] 的张量创建维度为 [N,D] 的矩阵

Question

我有一个维度为 N 的张量，我想复制它以创建一个维度为 NxD 的张量，每一列都是初始向量。

谢谢

Answer 1

您希望首先将张量扩展/重塑为 N x 1 形状，然后在第二维中将其平铺 D 次：

tensor_N_x_1 = tf.expand_dims(tensor, 1)     # Expand by adding a dim in position 1
tensor_N_x_D = tf.tile(tensor_N_x_1, [1, D]) # Tile 1 time in the 1st dim, D times in the 2nd

文档：

• tf.expand_dims
• tf.tile