How do I fix the UserWarning: Using a target size (torch.Size([1])) that is different to the input size (torch.Size([])) is deprecated

Question

When I try to run my model, I get this UserWarning. How do I go about fixing this? My model\'s input is expecting (torch.Size([])), a scalar, but my target is torch.Size([1]