tensorflow如何获取模型所有参数总和数量-创新互联

这篇文章给大家分享的是有关tensorflow如何获取模型所有参数总和数量的内容。小编觉得挺实用的，因此分享给大家做个参考，一起跟随小编过来看看吧。

成都创新互联公司主营北京网站建设的网络公司,主营网站建设方案,app软件开发,北京h5成都小程序开发搭建,北京网站营销推广欢迎北京等地区企业咨询

实例如下所示：

from functools import reduce
from operator import mul

def get_num_params():
 num_params = 0
 for variable in tf.trainable_variables():
  shape = variable.get_shape()
  num_params += reduce(mul, [dim.value for dim in shape], 1)
 return num_params

感谢各位的阅读！关于“tensorflow如何获取模型所有参数总和数量”这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，让大家可以学到更多知识，如果觉得文章不错，可以把它分享出去让更多的人看到吧！

另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。

当前题目：tensorflow如何获取模型所有参数总和数量-创新互联
URL分享：https://www.cdcxhl.com/article38/cddgpp.html

成都网站建设公司_创新互联，为您提供域名注册、网站维护、手机网站建设、动态网站、面包屑导航、搜索引擎优化

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源：创新互联

猜你还喜欢下面的内容