tensorflow统计graph中的trainable_variables

比眉伴天荒 2022-05-19 03:34 174阅读 0赞

最简单的做法: 转自: [https://blog.csdn.net/feynman233/article/details/79187304][https_blog.csdn.net_feynman233_article_details_79187304], 版权归原作者所有。

print(np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()]))

另有篇博客讲解的很详细:原文地址[https://blog.csdn.net/shwan\_ma/article/details/78879620][https_blog.csdn.net_shwan_ma_article_details_78879620]，版权归原作者所有。

原博主写的很好，将常用的方法记载下来供以后学习参考。

sess.run(tf.global\_varibales\_initializer())

variable\_name = \[v.name for v in tf.trainable\_variables()\]

print(variable\_names)

variable\_names = \[v.name for v in tf.trainable\_variables()\]

values = sess.run(variable\_names)

for k,v in zip(variable\_names, values):

print("Variable: ", k)

print("Shape: ", v.shape)

print(v)

for variable in tf.trainable\_variables():

shape = variable.get\_shape()

variable\_parameters = 1

for dim in shape:

variable\_parameters \*= dim.value

total\_parameters += variable\_parameters

[https_blog.csdn.net_feynman233_article_details_79187304]: https://blog.csdn.net/feynman233/article/details/79187304
[https_blog.csdn.net_shwan_ma_article_details_78879620]: https://blog.csdn.net/shwan_ma/article/details/78879620

发表评论取消回复

表情：

评论列表（有 0 条评论，174人围观）

还没有评论，来说两句吧...

相关阅读

相关【Tensorflow】tf.import_graph_def

tf.import\_graph\_def `tf.import_graph_def( graph_def, input_map=None, return_eleme...

小咪咪/ 2024年04月17日 16:46/ 0 赞/ 61 阅读

相关 module 'tensorflow' has no attribute 'get_default_graph'

参考：https://blog.csdn.net/u014466109/article/details/88877321?depth\_1-utm\_source=distri

末蓝、/ 2023年07月15日 13:06/ 0 赞/ 43 阅读

相关 tensorflow生成常量_TensorFlow 之计算图Graph的构建和执行

计算图(Graph)用来表示计算任务。图中的节点称为op(operation)。计算图必须在会话中启动，会话会把op分发到cpu或gpu等计算设备上进行执行，执行后会返

￡神魔★判官ぃ/ 2023年01月10日 05:09/ 0 赞/ 147 阅读

相关 tensorflow 出现ImportError: cannot import name graph_util

from tensorflow.python.framework import graph\_util 此句出现错误 ImportError: cannot impo

拼搏现实的明天。/ 2022年06月03日 07:39/ 0 赞/ 195 阅读

相关 tensorflow 获取graph中的所有tensor

[n.name for n in tf.get_default_graph().as_graph_def().node] 摘自[https://stackoverfl

我就是我/ 2022年05月19日 04:49/ 0 赞/ 202 阅读

相关 tensorflow统计graph中的trainable_variables

最简单的做法: 转自: [https://blog.csdn.net/feynman233/article/details/79187304][https_blog.csdn.

比眉伴天荒/ 2022年05月19日 03:34/ 0 赞/ 175 阅读

相关【tensorflow】Graph对象、Session对象、占位符

学习笔记TF005:Graph对象、Session对象、占位符 2017.05.10 03:07\ 字数 736 阅读 167 评论 0 喜欢 0 创建Graph对象，

拼搏现实的明天。/ 2022年05月18日 06:23/ 0 赞/ 184 阅读

相关 tensorflow打印graph

参考：[https://blog.csdn.net/cooco369/article/details/82999172 ][https_blog.csdn.net_cooco3

今天药忘吃喽~/ 2022年04月10日 13:15/ 0 赞/ 308 阅读

相关 TensorFlow为什么要使用Graph？

主要有以下几点原因：（1）节省资源、高效运算我么只需要运算需要的依赖项，不需要的不进行运算。（2）把整个运算分解成子环节，方便自动求导；（3）对分布式运算友好，

Dear 丶/ 2021年12月19日 00:09/ 0 赞/ 238 阅读

相关 tensorflow不知代码情况下获取graph的placeholder和output

Here is my way to find out the correct placeholder op name in the Graphdef part of th...

ゝ一纸荒年。/ 2021年05月19日 15:21/ 0 赞/ 348 阅读