tf.experimental.async_scope

Менеджер контекста для группировки асинхронных операций.

@tf_contextlib.contextmanager
tf.experimental.async_scope()

Вызовы операций/функций внутри области могут возвращаться до завершения фактического выполнения. При выходе из асинхронной области автоматически добавляется барьер синхронизации для обеспечения завершения всех асинхронных вычислений операций и функций, потенциально порождая исключения, если асинхронное выполнение приводит к ошибочному состоянию.

Пользователи могут написать следующий код для асинхронного вызова train_step_fn и записи метрики loss для каждых num_steps шагов в цикле обучения. train_step_fn внутренне потребляет данные с помощью iterator.get_next() и может вызывать OutOfRangeError при исчерпании данных. В случае:

try:
  with tf.experimental.async_scope():
    for _ in range(num_steps):
      # Step function updates the metric `loss` internally
      train_step_fn()
except tf.errors.OutOfRangeError:
  tf.experimental.async_clear_error()
logging.info('loss = %s', loss.numpy())