先构造一个简单的计算图,然后通过Tensorboard来查看图结构。
# -*- coding: utf-8 -*-
"""
Created on Mon Apr 1 21:20:40 2019
@author: Elyod
"""
import tensorflow as tf
x = tf.constant(1.0, name="input")
w = tf.Variable(0.5, name="weight")
b = tf.Variable(0.1, name="bias")
y = tf.add(tf.multiply(x,w, name="mul_op"), b, name="add_op")
# 设置写入的文件夹
summary_writer = tf.summary.FileWriter('./calc_graph')
# 获取默认的图
graph = tf.get_default_graph()
summary_writer.add_graph(graph)
# 将图结构写入文件
summary_writer.flush()
运行代码后,会将图结构写入到calc_graph文件夹的event文件里,按上面的代码,calc_graph文件夹和代码文件在同一个目录下。
打开Anaconda Prompt,进入到上述目录下,激活tensorflow环境,命令行输入
tensorboard --logdir=calc_graph
根据输出结果,打开Google浏览器输入http://L:6006,就会看到下图的结果了。
上图的结构中可以看到,一个名为input的常量和一个名为weight的变量相乘后,还会再与一个名为bias的变量相加(定义一个操作时,给操作的name参数赋值,会在这里显示出来)。当鼠标放到weight和bias上时,会出现可以点开的+号,点开后,里面有4个操作,这是因为一个变量包含4个操作,点击其中任意一个操作,在右边可以看到这个操作的详细信息。
参考:《Tensorflow入门与实战》罗冬日 人民邮电出版社