Lesson 1 - MSE function part - numel() method

Hi, the lecture video showed “torch.sum(diff * diff) / diff.numel()” as part of the MSE function when it was being defined. my question is: is diff.numel() the same as torch.numel(diff) ? because that is how it is written in the documentation.

2 Likes

Yes I believe it would be the same.

2 Likes

Both are same and supported by PyTorch.image

1 Like