pr = torch.tensor([[1.0], [2.0], [3.0], [4.0]])   # (4, 1)
tg = torch.tensor([1.0, 2.0, 3.0, 4.0])           # (4,)

print(nn.MSELoss()(pr, tg))     # tensor(2.5000)