TensorFlow计算图可视化

# -*- coding: utf-8 -*-

import tensorflow as tf
a=tf.constant(5,name='input_a')
b=tf.constant(3,name='input_b')
c=tf.multiply(a,b,name='maltiply_c')
d=tf.add(a,b,name='add_d')
e=tf.add(d,c,name='add_e')
sess=tf.Session()
sess.run(e)
#加上这行代码，就可以可视化。指定文件的路径，没有会自动生成。指定数据源：计算图
#将计算图上面的数据写入events文件里面，tensorboard可以同时从events文件里面读取数据并显示在web端
 writer=tf.summary.FileWriter('./improved_graph2',sess.graph)#tf.summary.FileWriter返回一个FileWriter类，它有很多其他方法

    writer.flush()
    writer.close()

    sess.close()

============================code=end=======

我们代码里面指定了文件的相对与当前代码的同级目录下，会生成一个events.out.xx的文件。

在终端进入该文件路径下输入

然后再IE上输入你远程主机的 IP地址：6009.就可以看到如下