我有一个B x 1形状的numpyTensor,名为Y。我在Pytorch中寻找下面展开命令的numpy等价形式(N是整数):
Y
N
Y.expand(-1, N)[:, None]
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]])
1条答案
按热度按时间qni6mghb1#
那瓷砖呢?