torch.sum的用法

import torch
a = torch.randn(3,5)
a
Out[4]: 
tensor([[ 1.9081e-01,  1.8261e+00,  1.1623e+00, -5.6293e-01, -1.1898e-01],
        [-1.8964e-01,  2.0439e-01, -2.2664e+00,  2.1306e-03, -2.0576e-02],
        [ 1.2379e+00,  7.6757e-01, -1.5163e+00, -1.9007e+00,  1.7272e-01]])
torch.sum(a)
Out[5]: tensor(-1.0117)
torch.sum(a,0)
Out[6]: tensor([ 1.2391,  2.7980, -2.6205, -2.4615,  0.0332])
torch.sum(a,1)
Out[7]: tensor([ 2.4972, -2.2701, -1.2388])
torch.sum(a,1,True)# True  保留原始维度
Out[9]: 
tensor([[ 2.4972],
        [-2.2701],
        [-1.2388]])
torch.sum(a,1,True).shape
Out[10]: torch.Size([3, 1])
torch.sum(a,1).shape
Out[11]: torch.Size([3])