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 相关文章推荐
探索Python3.4中新引入的asyncio模块
Apr 08 Python
在Python下进行UDP网络编程的教程
Apr 29 Python
python3中set(集合)的语法总结分享
Mar 24 Python
Python安装lz4-0.10.1遇到的坑
May 20 Python
一行python实现树形结构的方法
Aug 09 Python
pytorch 固定部分参数训练的方法
Aug 17 Python
python爬虫 2019中国好声音评论爬取过程解析
Aug 26 Python
使用django和vue进行数据交互的方法步骤
Nov 11 Python
Django扫码抽奖平台的配置过程详解
Jan 14 Python
Pytest之测试命名规则的使用
Apr 16 Python
Python实现智慧校园自动评教全新版
Jun 18 Python
python中Pyqt5使用Qlabel标签播放视频
Apr 22 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数组函数序列之array_search()- 按元素值返回键名
2011/11/04 PHP
php实现生成code128条形码的方法详解
2017/07/19 PHP
JS网络游戏-(模拟城市webgame)提供的一些例子下载
2007/10/14 Javascript
jquery的flexigrid无法显示数据提示获取到数据
2013/07/19 Javascript
JavaScript的模块化:封装(闭包),继承(原型) 介绍
2013/07/22 Javascript
jquery 合并内容相同的单元格(示例代码)
2013/12/13 Javascript
flash遮住div问题的正确解决方法
2014/02/27 Javascript
详解AngularJS实现表单验证
2015/12/10 Javascript
利用iscroll4实现轮播图效果实例代码
2017/01/11 Javascript
微信小程序页面滑动屏幕加载数据效果
2020/11/16 Javascript
JQuery模拟实现网页中自定义鼠标右键菜单功能
2018/11/14 jQuery
微信小程序页面调用自定义组件内的事件详解
2019/09/12 Javascript
小程序实现左滑删除的效果的实例代码
2020/10/19 Javascript
[01:57]DOTA2上海特锦赛小组赛解说单车采访花絮
2016/02/27 DOTA
[00:12]DAC SOLO赛卫冕冠军 VG.Paparazi灬展现SOLO技巧
2018/04/06 DOTA
Python读写Redis数据库操作示例
2014/03/18 Python
Python注释详解
2016/06/01 Python
Jupyter notebook远程访问服务器的方法
2018/05/24 Python
Python基于多线程实现抓取数据存入数据库的方法
2018/06/22 Python
Python实现数据结构线性链表(单链表)算法示例
2019/05/04 Python
Python3离线安装Requests模块问题
2019/10/13 Python
利用python Selenium实现自动登陆京东签到领金币功能
2019/10/31 Python
使用IPython或Spyder将省略号表示的内容完整输出
2020/04/20 Python
Python ini文件常用操作方法解析
2020/04/26 Python
Python尾递归优化实现代码及原理详解
2020/10/09 Python
Python Selenium库的基本使用教程
2021/01/04 Python
css3media响应式布局实例
2016/07/08 HTML / CSS
HTML5 canvas基本绘图之文字渲染
2016/06/27 HTML / CSS
Harman Audio官方商店:购买JBL、Harman Kardon、Infinity和AKG
2019/12/05 全球购物
文员自我评价怎么写
2013/09/19 职场文书
迟到早退检讨书
2014/02/10 职场文书
服装设计专业毕业生求职信
2014/04/09 职场文书
幸福中国演讲稿
2014/09/12 职场文书
2014年团总支工作总结
2014/11/21 职场文书
统计员岗位职责
2015/02/11 职场文书
在Java中Collection的一些常用方法总结
2021/06/13 Java/Android