如何可视化张量流 2 的张量板

Question

我是新手，在使用 tensorboard 时遇到了一些麻烦。 我在 Anaconda 提示符下盯着 Spyder 看

conda activate D:\Software\Anaconda\envs\tf
spyder

这是我在文件 trial.py 中的简单代码

import tensorflow as tf
a = 2
b = 3
x = tf.add(a, b)
writer = tf.summary.create_file_writer('./graphs')
print(writer)

然后我在 spyder 中运行了这段代码。我打开了另一个 Anaconda 提示并添加了这一行

conda activate D:\Software\Anaconda\envs\tf
tensorboard --logdir=D:\Dung\Maytinh\Python\graphs --port 6006

我打开 Chrome 并输入：http://localhost:6006/。结果如下：

我尝试了很多次，我无法理解错误在哪里。网上有些教程属于tensorflow 1，不适合处理。请帮帮我！

Answer 1

这是基于documentation

import tensorflow as tf
a = 2
b = 3

@tf.function
def func(x, y):
  return tf.add(a, b)

writer = tf.summary.create_file_writer('./graphs')

tf.summary.trace_on(graph=True, profiler=True)
z = func(a, b)
with writer.as_default():
  tf.summary.trace_export(
      name="trace",
      step=0,
      profiler_outdir='./graphs')

tensorboard --logdir ./graphs