tf.summary.record_if

Включает или выключает запись сводок в зависимости от предоставленного булевого значения.

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

Предоставленное значение может быть булевым значением Python, скалярным булевым тензором или вызываемым объектом, предоставляющим такое значение; если передан вызываемый объект, он будет вызываться по требованию, чтобы определить, будет ли происходить запись сводок. Обратите внимание, что при вызове record_if() в контексте режима eager, если вы планируете предоставить изменяющееся условие, например step % 100 == 0, вы должны обернуть его в вызываемый объект, чтобы избежать немедленного вычисления условия в режиме eager. В частности, использование вызываемого объекта — единственный способ вычислить ваше условие как часть отслеживаемого тела @tf.function, который вызывается из record_if() контекста.