tensorboard 展示 tensorflow的gragh
要在TensorBoard中展示TensorFlow的计算图,需要使用tf.summary.FileWriter将计算图写入磁盘,并使用tf.summary.FileWriter.add_graph将其添加到TensorBoard中。以下是一个简单的示例:
import tensorflow as tf
# 定义计算图
x = tf.placeholder(tf.float32, name='x')
y = tf.placeholder(tf.float32, name='y')
z = tf.add(x, y, name='z')
# 创建一个tf.summary.FileWriter对象
writer = tf.summary.FileWriter('./logs', tf.get_default_graph())
# 将计算图写入磁盘
writer.add_graph(tf.get_default_graph())
# 关闭writer
writer.close()
运行此代码后,将在"./logs"目录下生成一个事件文件。然后,使用以下命令启动TensorBoard:
tensorboard --logdir=./logs
在浏览器中打开指定的URL,您将在"Graphs"选项卡下看到您的计算图
原文地址: http://www.cveoy.top/t/topic/h00v 著作权归作者所有。请勿转载和采集!