Python的内置函数(四十二)、 numel()

112 阅读 0 评论 74 点赞

我是靠谱客的博主温暖皮带，这篇文章主要介绍Python的内置函数(四十二)、 numel()，现在分享给大家，希望可以做个参考。

参考 python中的numel()函数 - 云+社区 - 腾讯云

numel()函数：返回数组中元素的个数。

例：

复制代码

1
2
3
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的内置函数(四十二)、内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。