如何查看Tensorflow的graph

要查看 TensorFlow 的 session graph 結(jié)構(gòu),可以使用以下步驟:

創(chuàng)建一個 TensorFlow 的?Session?對象。

import? tensorflow as tf

# 創(chuàng)建一個 TensorFlow 的 Session 對象

sess = tf.Session()

使用?tf.get_default_graph()?函數(shù)獲取默認的 graph。

# 獲取默認的 graph

graph = tf.get_default_graph()

使用?graph.as_graph_def()?函數(shù)將 graph 轉(zhuǎn)換為?GraphDef?對象。

# 將 graph 轉(zhuǎn)換為 GraphDef 對象

graph_def = graph.as_graph_def()

使用?graph_def?對象的?node?屬性可以查看 graph 中的所有節(jié)點。

# 遍歷所有節(jié)點

for node in graph_def.node:

????print(node.name)

通過以上步驟,可以查看 TensorFlow 的 session

?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容