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í)行”，請(qǐng)使用默認(rèn)圖形。
• run_meta：可選的tensorflow.RunMetadata proto。有必要支持運(yùn)行時(shí)信息分析，例如時(shí)間和內(nèi)存。
• op_log：tensorflow.tfprof.OpLogProto proto。用戶可以使用op_log為圖節(jié)點(diǎn)分配“types”?！皌ypes”允許用戶使用選項(xiàng)['accounts_type_regexes']靈活地分組和帳戶配置文件。
• cmd：字符串?？梢允牵?#39;op'，'scope'，'graph'或'code'。'op'視圖使用操作類(lèi)型組織profile。（例如，MatMul）“scope”視圖使用圖形節(jié)點(diǎn)名稱(chēng)范圍組織profile。“graph”視圖使用圖形節(jié)點(diǎn)輸入/輸出組織profile。'code'視圖使用Python調(diào)用堆棧組織profile。
  
• options：選項(xiàng)的一個(gè)詞。

返回：

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