Python tf.experimental.async_clear_error用法及代碼示例

清除異步執行中的掛起操作和錯誤狀態。

用法

tf.experimental.async_clear_error()

在異步執行模式下，操作/函數執行中的錯誤可能會導致後續已計劃但尚未執行的操作/函數發生錯誤。調用此方法會清除所有掛起的操作並重置異步執行狀態。

例子：

while True:
  try:
    # Step function updates the metric `loss` internally
    train_step_fn()
  except tf.errors.OutOfRangeError:
    tf.experimental.async_clear_error()
    break
logging.info('loss = %s', loss.numpy())