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

Question

我有一个张量out，其形状为：

torch.Size([4, 644, 25])

和另一个x，形状为：

torch.Size([4, 161, 25])

我想将x添加到out 4次，例如：

        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的{​​{3}}：

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