在numpy中展开向量(Pytorch expand)

snz8szmq  于 2022-11-29  发布在  其他
关注(0)|答案(1)|浏览(130)

我有一个B x 1形状的numpyTensor,名为Y。我在Pytorch中寻找下面展开命令的numpy等价形式(N是整数):

Y.expand(-1, N)[:, None]
qni6mghb

qni6mghb1#

那瓷砖呢?

>>> B = 4
>>> N = 3
>>> a = np.arange(0, B).reshape(B, 1)
>>> np.tile(a, (1, N))
array([[0, 0, 0],
       [1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])

相关问题