Как отобразить базовый график тензорного потока с помощью трассировки_on() и трассировки_export() с помощью тензорной до ⇐ Python

[Anonymous]

Я хотел бы изучить тензорный поток. Хотя, когда я запускал код в своей Google-Colab, у меня появилось сообщение об ошибке:
'UnavailableError: Невозможно экспортировать результаты профилирования. Ни один профайлер не запущен.
Код взят с веб-сайта:

https://www.tensorflow.org/tensorboard/graphs
Подробности кода:

Код: Выделить всё

%load_ext tensorboard

import tensorflow as tf
from datetime import datetime

# The function to be traced
@tf.function
def my_func(x, y):
r = tf.matmul(x, y)
return r

# Set up logging
stamp = datetime.now().strftime("%Y%m%d-%H%M%S")
logdir = 'logs/func/%s' % stamp
writer = tf.summary.create_file_writer(logdir)

# Sample data for the 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)

%tensorboard --logdir logs/func

Сообщение об ошибке приведено ниже:



Я пытался поиграть с 'profiler=True' и 'profiler_outdir=logdir' в соответствии с рекомендацией, но ни один из них не помог. Может ли кто-нибудь помочь мне в этом? Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/798 ... -export-by