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

我有一个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

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


红糖糍粑
浏览 82回答 1
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)
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python