我是靠谱客的博主 温暖皮带,最近开发中收集的这篇文章主要介绍Python的内置函数(四十二)、 numel(),觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

参考  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()所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

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

评论列表共有 0 条评论

立即
投稿
返回
顶部