Tensorboard 的简单使用

TensorFlow与TensorBoard入门

最新推荐文章于 2024-06-26 15:46:35 发布

转载最新推荐文章于 2024-06-26 15:46:35 发布 · 64 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/llfctt/p/9617779.html

本文介绍如何使用TensorFlow创建简单向量加法运算，并利用TensorBoard进行可视化展示。通过实例演示了TensorFlow中常量、变量及操作的定义过程，并展示了如何启动TensorBoard查看计算图。

确保环境以及安装好tensorflow以及tensorboard

下面通过一个简单的例子来显示一下使用方式，一个向量加法的图结构。

import tensorflow as tf
a = tf.constant([1.0,2.0,3.0],name='input1')
b = tf.Variable(tf.random_uniform([3]),name='input2')
add = tf.add_n([a,b],name='addOP')
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    writer = tf.summary.FileWriter(“TensorBoard/test",sess.graph)
    print(sess.run(add))
writer.close()

运行完程序后，图结构将以日志文件的形式保存到给定的路径下。然后在命令行启动Tensorboard。Tensorboard作为本地的一个服务已经启动，端口号是6006，我们只需要在浏览器下输入127.0.0.1:6006就可以进入TensorBoard。下图分别是Tensorboard的界面，以及上面代码对用的图结构。启动命令：