概述
参考 python中的numel()函数 - 云+社区 - 腾讯云
numel()函数:返回数组中元素的个数。
例:
params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel()
print('#Params: %.1fM' % (params))
net.parameters():是Pytorch用法,用来返回net网络中的参数
params:用来返回net网络中的参数的总数目
最后
以上就是温暖皮带为你收集整理的Python的内置函数(四十二)、 numel()的全部内容,希望文章能够帮你解决Python的内置函数(四十二)、 numel()所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复