TensorBoard生成计算图

TensorBoard 生成计算图

这是一个简单的全连接网络&＃xff0c;实现a&＃61;x * w1 &＃43; b1, y&＃61;a * w2 &＃43; b2.
只看代码不是很形象&＃xff0c;计算图让网络变得更直观。
用 tf.summary.FileWriter() 方法生成计算图。
步骤&＃xff1a;
1&＃xff0c;运行代码&＃xff0c;把计算图保存在 "/home/yuejian/Desktop/graph"

import tensorflow as tf
x &＃61; tf.constant([0.9, 0.85], shape&＃61;[1, 2])
w1 &＃61; tf.Variable(tf.constant([[0.2, 0.1, 0.3], [0.2, 0.4, 0.3]], shape&＃61;[2, 3]), name&＃61;&＃39;w1&＃39;)
w2&＃61;tf.Variable(tf.constant([0.2, 0.5,0.25], shape&＃61;[3, 1]), name&＃61;&＃39;w2&＃39;)
b1&＃61;tf.constant([-0.3, 0.1, 0.2], shape&＃61;[1, 3], name&＃61;"b1")
b2&＃61;tf.constant([-0.3], shape&＃61;[1], name&＃61;"b2")init_op&＃61;tf.global_variables_initializer()
a&＃61;tf.matmul(x, w1)&＃43;b1
y&＃61;tf.matmul(a, w2)&＃43;b2
with tf.Session() as sess:sess.run(init_op)writer &＃61; tf.summary.FileWriter("/home/yuejian/Desktop/graph", sess.graph) #生成计算图writer.close()print(sess.run(y))


2&＃xff0c;在文件夹下找到刚刚保存好的计算图

3&＃xff0c;在当前文件夹下右键打开命令行
&＃xff08;注意要切换到刚刚运行程序的环境&＃xff09;
输入 tensorboard --logdir&＃61;/home/yuejian/Desktop/graph

4&＃xff0c;打开弹出的网址&＃xff0c; 比如我这里是 http://0.0.0.0:6006

5&＃xff0c;进入网址&＃xff0c;点击GRAPHS

6&＃xff0c;点击左上角下载&＃xff0c;将计算图保存到本地&＃xff0c;打开。

结束