torch.sum()

import torch

a = torch.randint(0,3,(2,3,4))
print(a.size())
print(a)
b = torch.sum(a, 0) # 按将第一个纬度的值相加，会减少一个纬度
c = torch.sum(a, 1) # 按将第二个纬度的值相加，会减少一个纬度
print('b', b)
print('c', c)