参考:https://blog.csdn.net/schmiloo/article/details/107020922
统计tensor张量包含的元素个数
import torch
x = torch.randn(3,3)
print("number elements of x is ",x.numel())
y = torch.randn(3,10,5)
print("number elements of y is ",y.numel())
输出内容:
number elements of x is 9
number elements of y is 150