tf.compat.v1.executing_eagerly

Проверяет, включено ли немедленное выполнение в текущем потоке.

tf.compat.v1.executing_eagerly()

Немедленное выполнение обычно включается с помощью tf.compat.v1.enable_eager_execution, но его также можно включить в контексте функции Python с помощью tf.contrib.eager.py_func.

Если немедленное выполнение включено, в большинстве случаев возвращается True. Однако этот API может возвращать False в следующих случаях.

• Выполнение внутри tf.function, если только предварительно не вызван tf.init_scope или tf.config.run_functions_eagerly(True).
• Выполнение внутри функции преобразования для tf.dataset.
• Вызван tf.compat.v1.disable_eager_execution().

tf.compat.v1.enable_eager_execution()

Общий случай:

print(tf.executing_eagerly())
True

Внутри tf.function:

@tf.function
def fn():
  with tf.init_scope():
    print(tf.executing_eagerly())
  print(tf.executing_eagerly())
fn()
True
False

Внутри tf.function после вызова tf.config.run_functions_eagerly(True):

tf.config.run_functions_eagerly(True)
@tf.function
def fn():
  with tf.init_scope():
    print(tf.executing_eagerly())
  print(tf.executing_eagerly())
fn()
True
True
tf.config.run_functions_eagerly(False)

Внутри функции преобразования для tf.dataset:

def data_fn(x):
  print(tf.executing_eagerly())
  return x
dataset = tf.data.Dataset.range(100)
dataset = dataset.map(data_fn)
False