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