Hello, I was trying to get the total pixel count for an image tensor.
The only solution I found is torch.Tensor(np.prod(tensor.size())) which isn’t as elegant as I would like it to be.
Is there somewhere in the documentary I overlook that contains a way to directly return the value? If not, will it be useful if I make a PR about this?
Cheers.