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 的跟踪主体的一部分的唯一方法。