在调试时,如何打印Tensorflow中可训练的所有变量(列表格式)?

时间:2016-07-21 17:54:20

标签: python tensorflow

在调试时,如何打印Tensorflow中可训练的所有变量(列表格式)?

例如,

    tvars = tf.trainable_variables()

我想检查tvars中的所有变量(列表类型)。

我已经尝试过以下代码,它会返回错误,

    myvars = session.run([tvars])
    print(myvars)

2 个答案:

答案 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复制了此内容。