在调试时,如何打印Tensorflow中可训练的所有变量(列表格式)?
例如,
tvars = tf.trainable_variables()
我想检查tvars中的所有变量(列表类型)。
我已经尝试过以下代码,它会返回错误,
myvars = session.run([tvars])
print(myvars)
答案 0 :(得分:13)
由于tf.trainable_variables()
会返回tf.Variable
个对象的列表,您应该能够将其结果直接传递给Session.run()
:
tvars = tf.trainable_variables()
tvars_vals = sess.run(tvars)
for var, val in zip(tvars, tvars_vals):
print(var.name, val) # Prints the name of the variable alongside its value.
答案 1 :(得分:3)
要打印张量流图的所有变量或节点的完整列表,您可以尝试这样做:
[n.name for n in tf.get_default_graph().as_graph_def().node]
我从here复制了此内容。