Python tf.summary.record_if用法及代碼示例

根據提供的布爾值打開或關閉摘要記錄。

用法

@tf_contextlib.contextmanager
tf.summary.record_if(
    condition
)

提供的值可以是 python 布爾值、標量布爾張量或提供此類值的可調用對象；如果傳遞了一個可調用對象，它將被調用on-demand 以確定是否會發生摘要寫入。請注意，在即刻模式上下文中調用 record_if() 時，如果您打算提供類似 step % 100 == 0 的變化條件，則必須將其包裝在可調用對象中以避免立即對條件進行即刻評估。特別是，使用可調用函數是將條件評估為從 record_if() 上下文中調用的 @tf.function 的跟蹤主體的一部分的唯一方法。