从维数为[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