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