как работать с оператором в python, тензофлоу 2.1.0?

Question

У меня сегодня глупый вопрос.

Я работаю над преобразованием некоторого кода, написанного для тензорного потока 1.x -> 2.1.0. Сегодня я столкнулся со случаем, когда мне нужно изменить способ использования summary.FileWriter.

На этой странице https://www.tensorflow.org/tensorboard/migrate я обнаружил, что теперь я хочу использовать create_file_writer и используйте оператор with, пример показан ниже.

writer = tf.summary.create_file_writer("/tmp/mylogs/eager")

with writer.as_default():
  for step in range(100):
    # other model code would go here
    tf.summary.scalar("my_metric", 0.5, step=step)
    writer.flush() 

Теперь, на мой вопрос из этого примера, кажется, что если строка tf.summary.scalar перемещается для вызова внутри другой области видимости, как, например, в другой функции. Результат должен быть одинаковым да или нет? Могу ли я попытаться преобразовать, просто инкапсулировав мой основной внутри с оператором writer.as_defaul ()?