tensorflow 获取模型所有参数总和数量的方法


Posted in Python onJune 14, 2018

实例如下所示:

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实现单词拼写检查
Apr 25 Python
python中字典(Dictionary)用法实例详解
May 30 Python
Python利用字典将两个通讯录文本合并为一个文本实例
Jan 16 Python
TensorFlow实现Softmax回归模型
Mar 09 Python
mac系统下Redis安装和使用步骤详解
Jul 09 Python
python 计算两个列表的相关系数的实现
Aug 29 Python
python GUI库图形界面开发之PyQt5输入对话框QInputDialog详细使用方法与实例
Feb 27 Python
Python3 filecmp模块测试比较文件原理解析
Mar 23 Python
jupyter notebook tensorflow打印device信息实例
Apr 20 Python
浅谈Python中的字符串
Jun 10 Python
只用50行Python代码爬取网络美女高清图片
Jun 02 Python
Pytorch中Softmax和LogSoftmax的使用详解
Jun 05 Python
tensorflow 获取变量&打印权值的实例讲解
Jun 14 #Python
利用python对Excel中的特定数据提取并写入新表的方法
Jun 14 #Python
Python基于最小二乘法实现曲线拟合示例
Jun 14 #Python
详解python之协程gevent模块
Jun 14 #Python
python 筛选数据集中列中value长度大于20的数据集方法
Jun 14 #Python
浅谈Tensorflow由于版本问题出现的几种错误及解决方法
Jun 13 #Python
tensorflow: 查看 tensor详细数值方法
Jun 13 #Python
You might like
PHP 命令行工具 shell_exec, exec, passthru, system详细使用介绍
2011/09/11 PHP
windows7下安装php的imagick和imagemagick扩展教程
2014/07/04 PHP
网站被恶意镜像怎么办 php一段代码轻松搞定(全面版)
2018/10/23 PHP
thinkphp5 + ajax 使用formdata提交数据(包括文件上传) 后台返回json完整实例
2020/03/02 PHP
关于PhpStorm设置点击编辑文件自动定位源文件的实现方式
2020/12/30 PHP
jQuery 开天辟地入门篇一
2009/12/09 Javascript
异步加载script的代码
2011/01/12 Javascript
JS.getTextContent(element,preformatted)使用介绍
2013/09/21 Javascript
用jquery中插件dialog实现弹框效果实例代码
2013/11/15 Javascript
jquery实现倒计时效果
2015/12/14 Javascript
JavaScript组件开发之输入框加候选框
2017/03/10 Javascript
妙用缓存调用链实现JS方法的重载
2018/04/30 Javascript
实例讲解v-if和v-show的区别
2019/01/31 Javascript
vue过滤器用法实例分析
2019/03/15 Javascript
javascript实现移动端触屏拖拽功能
2020/07/29 Javascript
[02:33]2014DOTA2 TI每日综述 LGD涉险晋级DK闯入胜者组
2014/07/14 DOTA
[01:34:42]NAVI vs EG 2019国际邀请赛小组赛 BO2 第二场 8.15
2019/08/17 DOTA
python复制与引用用法分析
2015/04/08 Python
python删除不需要的python文件方法
2018/04/24 Python
Python中的random.uniform()函数教程与实例解析
2019/03/02 Python
使用Python发现隐藏的wifi
2020/03/04 Python
python判断正负数方式
2020/06/03 Python
关于python中remove的一些坑小结
2021/01/04 Python
HTML5本地存储localStorage、sessionStorage基本用法、遍历操作、异常处理等
2014/05/08 HTML / CSS
使用placeholder属性设置input文本框的提示信息
2020/02/19 HTML / CSS
HTML5中的网络存储实现方式
2020/04/28 HTML / CSS
Under Armour美国官网:美国知名高端功能性运动品牌
2016/09/05 全球购物
英国奢侈品网站:MatchesFashion
2016/12/16 全球购物
女装和独特珠宝:Sundance Catalog
2018/09/19 全球购物
英国豪华装饰照明品牌的在线零售商:Inspyer Lighting
2019/12/10 全球购物
大学生就业求职信
2014/06/12 职场文书
节水口号标语
2014/06/19 职场文书
北京青年观后感
2015/06/15 职场文书
创业计划书之宠物店
2019/09/19 职场文书
Python Flask请求扩展与中间件相关知识总结
2021/06/11 Python
深入理解MySQL中MVCC与BufferPool缓存机制
2022/05/25 MySQL