如何让TensorBoard在图表中显示变量名称?

时间:2017-09-20 11:04:49

标签: tensorflow tensorboard

我关注an intro to the TensorFlow并注意到所有图形可视化都有节点名称,与相应变量的名称相同。但是当我尝试生成相同的可视化时,TensorBoard只是将操作类型用作名称。

例如,以下代码与简介中的相同

import tensorflow as tf

sess = tf.Session()

a = tf.placeholder(tf.float32)
b = tf.placeholder(tf.float32)
adder_node = a + b
add_and_triple = adder_node * 3.

tf.summary.FileWriter("D:\\TF", sess.graph)

print(sess.run(add_and_triple, {a: 3, b: 4.5}))

当我使用TensorBoard可视化图形时,请告诉我

TensorBoard result

虽然在介绍中就像这样

enter image description here

我找到了手动设置名称的方法 - 使用像这样的name参数

a = tf.placeholder(tf.float32, name = 'a')

但这似乎是很多工作并且破坏了所有的美丽

adder_node = a + b

是否有任何设置可自动生成?

1 个答案:

答案 0 :(得分:0)

对于 tf.Placeholders tf.Variables tf.Constants ,您可以使用参数 name 来为张量流中的那些节点分配一个名称。这些名称将由TensorBoard在绘制图形时使用。

x = tf.placeholder(tf.float32, name="x")
y= tf.valiable(0.5, tf.float32, name="y")
z= tf.constant(1.0, tf.float32, name="z")

我认为没有一种自动的方法可以将代码中的变量名分配给计算图中的占位符和变量。