在火炬中的列总数

Question

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

例如： 一个：

1 2 3 4 5 6

我希望B

5 7 9

Answer 1

要做到这一点，你可以使用数总和法。

torch.sum(a,1)

在一般情况下，你可以指定要总结过任何轴。

torch.sum(a,axis)

（综上所述过的行，可以使用轴= 2）