使用tf.trainable_variables()显示可训练变量的名称
tensorflow
4
0

我正在处理RNNCell 权重

我有以下代码

var_names = []    
for var in tf.trainable_variables():    
    var_names.append(var.name)

在另一个名为model.py文件中,我正在打印仅附加到var_names的可训练变量的名称。但是,我发现可训练变量的“名称”属性没有用,因为它不是很具描述性。

RNNcell的权重RNNcell有名称?

如果有用的话,这是另一段可能相关的代码:

cell_fn = tf.nn.rnn_cell.GRUCell   
rnn_fw_1 = cell_fn(num_hidden_1, **additional_cell_args)    
rnn_fw_1 = tf.nn.rnn_cell.DropoutWrapper(rnn_fw_1, input_keep_prob=keep_prob_1)
参考资料:
Stack Overflow
收藏
评论
共 1 个回答
高赞 时间 活跃

尝试这个:

variables_names = [v.name for v in tf.trainable_variables()]
values = sess.run(variables_names)
for k, v in zip(variables_names, values):
    print "Variable: ", k
    print "Shape: ", v.shape
    print v
收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号