码迷,mamicode.com
首页 > 其他好文 > 详细

tf.trainable_variables() and tf.all_variables()

时间:2019-03-12 18:48:59      阅读:391      评论:0      收藏:0      [点我收藏+]

标签:glob   type   var   nbsp   span   variables   color   pre   [1]   

tf.trainable_variables()   返回的是 所有需要训练的变量列表

tf.all_variables()      返回的是 所有变量的列表

v = tf.Variable(0, name=v)
v1 = tf.Variable(tf.constant(5, shape=[1], dtype=tf.float32), name=v1)

global_step = tf.Variable(6, name=global_step, trainable=False)    # 声明不是训练变量

for ele1 in tf.trainable_variables():
    print(ele1.name)

print(**10)

for ele2 in tf.all_variables():
    print(ele2.name)

输出

v:0
v1:0
**********
v:0
v1:0
global_step:0

 

tf.trainable_variables() and tf.all_variables()

标签:glob   type   var   nbsp   span   variables   color   pre   [1]   

原文地址:https://www.cnblogs.com/yanshw/p/10518253.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!