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中的super如何使用
这篇文章主要介绍了python中的super,python中的super，名为超类，可以简单的理解为执行父类的__init__函数，本文就着重看下super的具体作用，需要的朋友可以参考下
2022-03-03
Python 3.6 -win64环境安装PIL模块的教程
PIL功能非常强大，但API却非常简单易用。这篇文章主要介绍了Python 3.6 -win64环境安装PIL模块的教程,需要的朋友可以参考下
2019-06-06
详解Python中键盘鼠标的相关操作
pyautogui模块，规范一点大写就是PyAutoGUI模块，是用来实现自动化脚本的一个十分得力的小助手。他可以操控键盘和鼠标，可以写许多工具。本文就来讲讲如何利用这个模块实现键盘鼠标的相关操作，需要的可以参考一下
2023-01-01
Python中join和split用法实例
这篇文章主要介绍了Python中join和split用法实例,本文直接给出使用实例,需要的朋友可以参考下
2015-04-04
python3.x实现发送邮件功能
这篇文章主要为大家详细介绍了python3.x实现发送邮件功能，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-05-05
疯狂上涨的Python 开发者应从2.x还是3.x着手?
热度疯涨的 Python，开发者应从 2.x 还是 3.x 着手？这篇文章就为大家分析一下了Python开发者应从2.x还是3.x学起，感兴趣的小伙伴们可以参考一下
2017-11-11
Python map及filter函数使用方法解析
这篇文章主要介绍了Python map及filter函数使用方法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-08-08
Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
今天小编就为大家分享一篇Pytorch 定义MyDatasets实现多通道分别输入不同数据方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01
python库patchworklib多图表整合用法示例探究
这篇文章主要介绍了python库patchworklib对齐matplotlib图表,将多个图表的整合为单一图表用法示例探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01
Python tornado上传文件的功能
这篇文章主要介绍了Python tornado上传文件的功能，代码分为普通上传和ajax上传，通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-03-03