tensorflow 获取模型所有参数总和数量的方法
更新时间：2018年06月14日 09:19:07 作者：guotong1988
今天小编就为大家分享一篇tensorflow 获取模型所有参数总和数量的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
实例如下所示：
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 获取模型所有参数总和数量的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
我们刷视频时常常会想把精彩的视频保存到本地，如果少数的还行，如果有很多的话一个个保存太麻烦了。本文教你如何用Python实现视频批量保存到本地，需要的可以参考一下2022-05-05
这篇文章主要介绍了Python中函数定义及其调用,感兴趣的朋友可以来了解一下2021-06-06
这篇文章主要介绍了Python带动态参数功能的sqlite工具类,涉及Python针对sqlite数据库的连接、查询、sql语句执行等相关操作封装与使用技巧,需要的朋友可以参考下2018-05-05
这篇文章主要介绍了python中的编码和解码及\x和\u问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
Python 统计列表中重复元素的个数并返回其索引值的实现方法
这篇文章主要介绍了Python 统计列表中重复元素的个数并返回其索引值,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-05-05
Python 实现 T00ls 自动签到脚本代码（邮件+钉钉通知）
这篇文章主要介绍了Python 实现 T00ls 自动签到脚本（邮件+钉钉通知）,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-07-07
这篇文章主要介绍了PyQt5/PySide2的按钮控件使用实例，代码简单易懂，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下2019-08-08
Django是基于MTV模式的框架，需要配合url控制器（路径分发）使用，本文重点给大家介绍Django MTV模式详解，需要的朋友参考下吧2021-09-09
这篇文章主要为大家介绍了python继续找对象的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-01-01
在很多数据处理项目中,将数据存储到数据库中是非常常见的操作,下面这篇文章主要给大家介绍了关于如何将python的数据存储到mysql数据库中的相关资料,需要的朋友可以参考下2023-12-12
最新评论