TensorFlow中的計算圖是一種數據流圖,用于描述數據流和操作之間的關系。計算圖由節點和邊組成,節點表示操作,邊表示數據流。
要使用計算圖,首先需要創建一個默認計算圖,可以使用tf.Graph()
來創建一個新的計算圖,也可以使用tf.get_default_graph()
來獲取默認計算圖。然后,可以使用with
語句來將操作添加到計算圖中。例如:
import tensorflow as tf
# 創建一個新的計算圖
graph = tf.Graph()
# 將操作添加到計算圖中
with graph.as_default():
a = tf.constant(3)
b = tf.constant(4)
c = a + b
# 創建一個會話
with tf.Session(graph=graph) as sess:
result = sess.run(c)
print(result) # 輸出:7
在上面的例子中,我們首先創建了一個新的計算圖,然后將常量a
和b
以及加法操作c
添加到計算圖中,并使用會話來執行計算圖中的操作。TensorFlow會自動構建計算圖并執行操作,最后輸出結果為7。