Python tf.executing_eagerly用法及代码示例

检查当前线程是否启用了即刻执行。

用法

tf.executing_eagerly()

默认情况下启用即刻执行，并且此 API 在大多数情况下返回 True。但是，在以下用例中，此 API 可能会返回 False。

• 在 tf.function 内部执行，除非之前调用了 tf.init_scope 或 tf.config.run_functions_eagerly(True)。
• 在 tf.dataset 的转换函数中执行。
• tf.compat.v1.disable_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.config.run_functions_eagerly(True) 之后的 tf.function 内部被调用：

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