pytorch獲取張量元素個數numel()的用法

 

numel就是"number of elements"的簡寫。numel()可以直接返回int類型的元素個數

1. import torch
2.  
3. a = torch.randn( 1, 2, 3, 4)
4. b = a.numel()
5. print(type(b)) # int
6. print(b) # 24

通過numel()函數，我們可以迅速查看一個張量到底又多少元素。