我可以添加两个张量x和y就地这样
x = x.add(y)
考虑到所有张量都具有相同的尺寸，是否可以对三个或更多张量进行相同的处理？
result = torch.sum(torch.stack([x, y, ...]), dim=0)
没有堆栈：
from functools import reduce
result = reduce(torch.add, [x, y, ...])
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句