python - Torch 沿轴对张量求和

Question

如何对张量的列求和？

torch.Size([10, 100])    --->    torch.Size([10])

Answer 1

最简单最好的解决方案是使用torch.sum()。

对一个张量的所有元素求和:

torch.sum(x) # gives back a scalar

对所有行求和(即每一列):

torch.sum(x, dim=0) # size = [ncol]

对所有列求和(即每一行):

torch.sum(x, dim=1) # size = [nrow]

需要注意的是，求和的维度会从结果张量中消除。