Tensorflow 笔记 -- tensorboard 的使用

Tensorflow 笔记 -- tensorboard 的使用

TensorFlow提供非常方便的可视化命令Tensorboard，先上代码

import tensorflow as tf

a = tf.constant(5, name="const_a")
b = tf.constant(4, name="const_b")

c = tf.multiply(a, b, name="mul_a_b")
d = tf.add(a, b, name="add_a_b")

e = tf.add(c, d, name="add_c_d")

with tf.Session() as sess:
    writer = tf.summary.FileWriter('graph', sess.graph)
    print(sess.run(e))
    writer.close()

首先是定义一张图，然后跑图。

这里需要注意一点是，FileWriter一定要在sess.run执行前先执行，也就是说，你得把writer = tf.summary.FileWriter('graph',sess.graph)写在sess.run(e)之前。参数graph表示的是你的logdir目录，后面需要使用。

运行完代码后，直接在代码所在的文件夹目录下执行命令tensorboard --logdir="graph"。然后会建立一个网站，在浏览器中打开得到的地址即可得到可视化界面。整个过程还是蛮简单的，就是TensorFlow这玩意的API变动的特别大，尤其是步入1.0版。。。较之前发生了比较大的变更。。。希望1.0出来后，API的变更不那么大了吧。。。。。。