火炬中的列总和

Question

我如何在火炬的柱子上总结？我有一个128*1024张量,我希望通过对所有行求和得到1*1024张量.

例如:a:

1 2 3 4 5 6

我想要b

5 7 9

Answer 1

为此,您可以使用sum方法.

torch.sum(a,1)

通常,您可以指定要求的任何轴.

torch.sum(a,axis)

(要对行求和,可以使用axis = 2)