import tensorflow as tf
@tf.function #启用图必须开启
def my_func(x, y):
# A simple hand-rolled layer.
return tf.multiply(x, y)
# Set up logging.
logdir = r"C:\Users\Admin\Documents\vscode\g"
writer = tf.summary.create_file_writer(logdir) #日志文件保存
# Sample data for your function.
x = tf.random.uniform((3, 3))
y = tf.random.uniform((3, 3))
# Bracket the function call with
# tf.summary.trace_on() and tf.summary.trace_export().
tf.summary.trace_on(graph=True, profiler=True) #打开流监控
# Call only one tf.function when tracing.
z = my_func(x, y)
with writer.as_default():
tf.summary.trace_export(
name="my_func_trace",
step=0,
profiler_outdir=logdir)
参考:
https://www.tensorflow.org/tensorboard/graphs?hl=zh-cn