torch.sum( dim)

torch.sum 에는 차원을 정해주는 argument가 있다.

dim=0 column 별로 더한 값이고

dim=1 은 row별로 더한 값을 출력한다.

import torch

a = torch.randn(2, 3)
>>>print(a)
>>>print(torch.sum(a, dim=0))
>>>print(torch.sum(a, dim=1))

tensor([[-0.2771,  0.4708,  0.7797],
        [ 2.8122, -1.0677,  1.7852]])
tensor([ 2.5351, -0.5969,  2.5649])
tensor([0.9734, 3.5298])