火炬中的列总和

Question

我如何沿火炬中的一列求和？我有一个 128*1024 的张量，我想通过对所有行求和得到一个 1*1024 的张量。

例如： 一个：

1 2 3 4 5 6

我想要b

5 7 9

Answer 1

为此，您可以使用 sum 方法。

torch.sum(a,1)

一般来说，您可以指定要求和的任何轴。

torch.sum(a,axis)

（要对行求和，可以使用axis=2）