torch.numel(input)-> int
returns the total number of elements in the input tensor
input tensor의 원소의 개수를 반환한다.
>>> a = torch.randn(1, 2, 3, 4, 5) >>> torch.numel(a) 120 >>> a = torch.zeros(4,4) >>> torch.numel(a) 16