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多继承的继承顺序,是按照深度遍历的方式,其实python多继承顺序的算法,不是严格意义上的深度遍历,而是基于深度遍历基础上优化出一种叫3C算法,本文将给大家详细的介绍一下python多继承的3C算法,需要的朋友可以参考下2024-07-07
这篇文章主要给大家介绍了关于python中round()函数用法的相关资料,round()函数是Python内置函数之一,用于对数字进行四舍五入操作,需要的朋友可以参考下2023-08-08
这篇文章主要介绍了用 Python 绘制全国鸿星尔克门店分布图，今天就以某度地图 用Python爬虫看一下全国到底有多少家鸿星尔克门店，,需要的朋友可以参考一下2022-01-01
Python中利用pyqt5制作指针钟表显示实时时间(指针时钟)
这篇文章主要介绍了Python中利用pyqt5制作指针钟表显示实时时间(动态指针时钟),本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-02-02
最近做程序用到汉字转汉语拼音的问题，首先想到Google一下，但是拿来看了一下源码，不是甚清楚。2009-02-02
这篇文章主要介绍了python实现交并比IOU教程，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
python利用os模块编写文件复制功能——copy()函数用法
这篇文章主要介绍了python利用os模块编写文件复制功能——copy()函数用法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
这篇文章主要介绍了Python中的groupby分组功能的实例代码，非常不错，具有一定的参考借鉴价值，需要的朋友参考下吧2018-07-07
Python自定义聚合函数merge与transform区别详解
这篇文章主要介绍了Python自定义聚合函数merge与transform区别详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-05-05
这篇文章主要介绍了python绘制铅球的运行轨迹代码分享，具有一定参考价值，需要的朋友可以了解下。2017-11-11
最新评论