numel就是"number of elements"的简写。numel()可以直接返回int类型的元素个数
import torcha = torch.randn(1, 2, 3, 4)
b = a.numel()
print(type(b)) # int
print(b) # 24
通过numel()函数,我们可以迅速查看一个张量到底又多少元素。
本文链接:https://my.lmcjl.com/post/3160.html
展开阅读全文
numel就是"number of elements"的简写。numel()可以直接返回int类型的元素个数
import torcha = torch.randn(1, 2, 3, 4)
b = a.numel()
print(type(b)) # int
print(b) # 24
通过numel()函数,我们可以迅速查看一个张量到底又多少元素。
本文链接:https://my.lmcjl.com/post/3160.html
4 评论