Python 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.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