a = torch.rand(2,4)
print(a)
print(a.exp())
print(a.exp().sum(-1))
print(a.exp().sum(-1).shape)
print(a.exp().sum(-1).unsqueeze(-1))
print(a.exp().sum(-1).unsqueeze(-1).shape)