[Python] How to find the number of elements in tensor?

Question

Which function should I use to count the number of elements in a given tensor?

Answer 1

You can use numel() function (number of elements) of the torch to find the number of elements in a given tensor.

Here is an example:

>>> import torch
>>> a=torch.tensor([1,2,3,4])
>>> a
tensor([1, 2, 3, 4])
>>> torch.numel(a)
4
>>> a=torch.randn(5,6)
>>> torch.numel(a)
30