在火炬中的列總數

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）