使用尺寸为[N]的张量创建尺寸为[N,D]的矩阵

时间:2018-04-07 16:40:12

标签: python python-3.x tensorflow

我有一个尺寸为N的张量,我想复制它以创建一个尺寸为NxD的张量,每列都是初始矢量。

谢谢

1 个答案:

答案 0 :(得分:3)

您希望首先将张量扩展/重塑为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

文档: