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

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

是否有一些我可以使用的单线来做到这一点？

标签： pythonnumpypytorch

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)