通过沿轴重复将张量添加到另一个

Question

我有一个out形状的张量：torch.Size([4, 644, 25])和另一个x形状：torch.Size([4, 161, 25])我想加到x4out次，比如：        out[:, 0:161] += x         out[:, 161:322] += x         out[:, 322:483] += x         out[:, 483:644] += x是否有一些我可以使用的单线来做到这一点？

Answer 1

我们可以np.tile在这里使用：

out += np.tile(x, (1,out.shape[1]//x.shape[1],1))

或使用pytorch's repeat：

out += x.repeat(1,out.shape[1]//x.shape[1],1)