Python tf.summary.SummaryWriter.as_default用法及代码示例

用法

as_default(
    step=None
)

参数

step int64 -castable 默认步长值，或 None 。如果不是 None ，则捕获当前步骤，用给定的步骤替换，并在上下文管理器退出时恢复原始步骤。当 None 时，当前步骤不会被修改(并且在上下文管理器退出时不会恢复)。

上下文管理器。

返回启用摘要编写的上下文管理器。

为方便起见，如果 step 不是 None，则此函数还会为 API 中其他位置的 summary-writing 函数中使用的 step 参数设置默认值，以便无需在每次此类调用中显式传递它。该值可以是常量或变量。

注意：在@tf.function 中设置step 时，将在跟踪函数时捕获步长值，因此除非使用tf.Variable 步长，否则对函数外步长的更改不会反映在函数内。

例如，step 可以用作：

with writer_a.as_default(step=10):
  tf.summary.scalar(tag, value)   # Logged to writer_a with step 10
  with writer_b.as_default(step=20):
    tf.summary.scalar(tag, value) # Logged to writer_b with step 20
  tf.summary.scalar(tag, value)   # Logged to writer_a with step 10

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.summary.SummaryWriter.as_default。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。