Python tf.profiler.experimental.Trace用法及代碼示例

在分析器中生成跟蹤事件的上下文管理器。

用法

tf.profiler.experimental.Trace(
    name, **kwargs
)

參數

name 跟蹤事件的名稱。

**kwargs 添加到跟蹤事件的關鍵字參數。 key 和 value 都是可以轉換為字符串的類型，分析器將根據 traceme 名稱對其進行解釋。

示例用法：

tf.profiler.experimental.start('logdir')
  for step in range(num_steps):
    # Creates a trace event for each training step with the
    # step number.
    with tf.profiler.experimental.Trace("Train", step_num=step):
      train_fn()
  tf.profiler.experimental.stop()

上麵的示例使用關鍵字參數 "step_num" 來指定要跟蹤的訓練步驟。

進入上下文時將啟動跟蹤事件，並在退出上下文時停止並將結果保存到分析器。打開 TensorBoard Profile 選項卡並選擇跟蹤查看器以查看時間軸中的跟蹤事件。

僅當啟用探查器時才會創建跟蹤事件。有關如何使用分析器的更多信息，請訪問 https://tensorflow.org/guide/profiler

示例用法：

tf.profiler.experimental.start('logdir')
for step in range(num_steps):
  # Creates a trace event for each training step with the step number.
  with tf.profiler.experimental.Trace("Train", step_num=step, _r=1):
    train_fn()
tf.profiler.experimental.stop()

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.profiler.experimental.Trace。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。