TensorBoard是TensorFlow的一个可视化工具。
可视化主要有两个流程,写入事件文件和可视化事件文件。
写入事件文件
TensorFlow 的事件文件描述了TensorFlow 运行中涉及到的主要数据。
可以给需要记录变化的节点添加summary op,让summary op来收集节点的数据,并标记上第几步、时间戳等标识,并写入到事件文件中。
import tensorflow as tf with tf.name_scope('input1'): input1 = tf.constant([1.0, 2.0, 3.0], name="input1") with tf.name_scope('input2'): input2 = tf.Variable(tf.random_uniform([3]), name="input2") output = tf.add_n([input1, input2], name="add") writer = tf.summary.FileWriter("/tmp/log", tf.get_default_graph()) writer.close()
可视化事件文件
运行tensorboard命令如下:
tensorboard --logdir=/tmp/log
打开http://127.0.0.1:6006/#graphs 网址,可以看到如下图所示的计算图结构,当然还可以查看其他数据,后面会做详细介绍。