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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。