pytorch获取张量元素个数numel()的用法

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()函数，我们可以迅速查看一个张量到底又多少元素。