火炬中的列总和

Column sum in torch

我如何沿着 torch 中的列求和?我有一个128*1024的张量,我想通过对所有行求和得到一个1*1024的张量。

例如: 一个:

1 2 3 4 5 6

我要b

5 7 9

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

torch.sum(a,1)

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

torch.sum(a,axis)

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