TensorFlow函數(shù)教程：tf.profiler.profile

tf.profiler.profile函數(shù)

tf.profiler.profile(
    graph=None,
    run_meta=None,
    op_log=None,
    cmd='scope',
    options=_DEFAULT_PROFILE_OPTIONS
)

定義在：tensorflow/python/profiler/model_analyzer.py。

profile模型。

可以在以下網(wǎng)址找到教程和示例：https：//github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler/README.md

參數(shù)：

• graph：tf.Graph。如果是“None”或者未啟用“eager執(zhí)行”，請使用默認圖形。
• run_meta：可選的tensorflow.RunMetadata proto。有必要支持運行時信息分析，例如時間和內存。
• op_log：tensorflow.tfprof.OpLogProto proto。用戶可以使用op_log為圖節(jié)點分配“types”?！皌ypes”允許用戶使用選項['accounts_type_regexes']靈活地分組和帳戶配置文件。
• cmd：字符串?？梢允牵?#39;op'，'scope'，'graph'或'code'。'op'視圖使用操作類型組織profile。（例如，MatMul）“scope”視圖使用圖形節(jié)點名稱范圍組織profile?！癵raph”視圖使用圖形節(jié)點輸入/輸出組織profile。'code'視圖使用Python調用堆棧組織profile。
  
• options：選項的一個詞。

返回：

如果cmd是'scope'或'graph'，則返回GraphNodeProto proto。如果cmd是'op'或'code'，則返回MultiGraphNodeProto proto。副作用：stdout/file/timeline.json取決于選項['output']。