Python tf.config.run_functions_eagerly用法及代碼示例

啟用/禁用 tf.function 的即刻執行。

用法

tf.config.run_functions_eagerly(
    run_eagerly
)

調用tf.config.run_functions_eagerly(True) 將使tf.function 的所有調用立即運行，而不是作為跟蹤圖函數運行。

這對於調試很有用。

def my_func(a):
 print("Python side effect")
 return a + a
a_fn = tf.function(my_func)

# A side effect the first time the function is traced
a_fn(tf.constant(1))
Python side effect
<tf.Tensor:shape=(), dtype=int32, numpy=2>

# No further side effect, as the traced function is called
a_fn(tf.constant(2))
<tf.Tensor:shape=(), dtype=int32, numpy=4>

# Now, switch to eager running
tf.config.run_functions_eagerly(True)
# Side effect, as the function is called directly
a_fn(tf.constant(2))
Python side effect
<tf.Tensor:shape=(), dtype=int32, numpy=4>

# Turn this back off
tf.config.run_functions_eagerly(False)

注意：此標誌對作為參數傳遞給 tf.data 轉換的函數沒有影響。 tf.data 函數永遠不會即刻地執行，並且始終作為已編譯的 Tensorflow Graph 執行。