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 相关文章推荐
centos 下面安装python2.7 +pip +mysqld
Nov 18 Python
python学习之第三方包安装方法(两种方法)
Jul 30 Python
python Selenium爬取内容并存储至MySQL数据库的实现代码
Mar 16 Python
python分块读取大数据,避免内存不足的方法
Dec 10 Python
对python实现合并两个排序链表的方法详解
Jan 23 Python
Flask框架学习笔记之路由和反向路由详解【图文与实例】
Aug 12 Python
pytorch 更改预训练模型网络结构的方法
Aug 19 Python
Python二次规划和线性规划使用实例
Dec 09 Python
将python依赖包打包成window下可执行文件bat方式
Dec 26 Python
Python内置类型性能分析过程实例
Jan 29 Python
Pandas时间序列:时期(period)及其算术运算详解
Feb 25 Python
python面向对象版学生信息管理系统
Jun 24 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取整的几种方式
2013/06/25 PHP
php把字符串指定字符分割成数组的方法
2018/03/12 PHP
Laravel 5.5 异常处理 & 错误日志的解决
2019/10/17 PHP
PHP连接MySQL数据库操作代码实例解析
2020/07/11 PHP
弹出广告特效代码(一个IP只弹出一次)
2007/05/11 Javascript
让js弹出窗口居前显示的实现方法
2013/07/10 Javascript
jQuery取id有.的值的方法
2014/05/21 Javascript
node.js中的http.response.writeHead方法使用说明
2014/12/14 Javascript
node.js中的fs.link方法使用说明
2014/12/15 Javascript
JavaScript中的类数组对象介绍
2014/12/30 Javascript
使用js画图之圆、弧、扇形
2015/01/12 Javascript
Javascript定义类(class)的三种方法详解
2015/03/13 Javascript
JS+CSS实现自动切换的网页滑动门菜单效果代码
2015/09/14 Javascript
ionic js 模型 $ionicModal 可以遮住用户主界面的内容框
2016/06/06 Javascript
Bootstrap标签页(Tab)插件使用方法
2017/03/21 Javascript
vuejs使用FormData实现ajax上传图片文件
2017/08/08 Javascript
AngularJS 将再发布一个重要版本 然后进入长期支持阶段
2018/01/31 Javascript
echarts统计x轴区间的数值实例代码详解
2019/07/07 Javascript
json 带斜杠时如何解析的实现
2019/08/12 Javascript
解决layui 三级联动下拉框更新时回显的问题
2019/09/03 Javascript
react native 仿微信聊天室实例代码
2019/09/17 Javascript
解决Layui 表格自适应高度的问题
2019/11/15 Javascript
使用python在校内发人人网状态(人人网看状态)
2014/02/19 Python
python代码制作configure文件示例
2014/07/28 Python
解读Python中degrees()方法的使用
2015/05/18 Python
使用Python的Tornado框架实现一个Web端图书展示页面
2016/07/11 Python
python使用邻接矩阵构造图代码示例
2017/11/10 Python
tensorflow指定GPU与动态分配GPU memory设置
2020/02/03 Python
CSS3让登陆面板3D旋转起来
2016/05/03 HTML / CSS
html5中为audio标签增加停止按钮动作实现方法
2013/01/04 HTML / CSS
世界顶级足球门票网站:Live Football Tickets
2017/10/14 全球购物
德国网上花店:Valentins
2018/08/15 全球购物
日本最大美瞳直送网:Morecontact(中文)
2019/04/03 全球购物
广播体操比赛口号
2014/06/10 职场文书
PyTorch 如何自动计算梯度
2021/05/23 Python
Django实现WebSocket在线聊天室功能(channels库)
2021/09/25 Python