要查看 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