import torch

m = torch.tensor([[1., 2., 3.],
                  [4., 5., 6.]])

print(m.sum(dim=1).shape)                 # torch.Size([2])
print(m.sum(dim=1, keepdim=True).shape)   # torch.Size([2, 1])