您好,登錄后才能下訂單哦!
本篇文章為大家展示了TensorFlow基礎中怎么創建計算圖,內容簡明扼要并且容易理解,絕對能使你眼前一亮,通過這篇文章的詳細介紹希望你能有所收獲。
下面介紹TensorFlow與創建計算圖的幾個函數:
g = tf.Graph() #創建新的計算圖g
g.as_default() #將計算圖g設置為當前使用的計算圖
g0 = tf.get_default_graph() # 獲取默認的計算圖
tf.reset_default_graph() #清空默認的計算圖
示范1:
import tensorflow as tf
# 初始化一個計算圖對象g
g = tf.Graph()
# 在g中添加節點定義計算圖
with g.as_default():
a = tf.constant(10,name = 'a')
b = tf.constant(5,name = 'b')
c = tf.add(a,b)
print(c.graph)
執行后輸出為:
示范2:
import tensorflow as tf
#獲取默認的計算圖對象句柄g0
g0 = tf.get_default_graph()
#在g0中添加節點定義計算圖
with g0.as_default():
a = tf.constant(10,name = 'a')
b = tf.constant(5,name = 'b')
c = tf.add(a,b)
print(c.graph)
執行后輸出為:
示范3:
import tensorflow as tf
#清空默認的計算圖
tf.reset_default_graph()
# 直接用Operator定義的節點將添加到默認計算圖中
a = tf.constant(10,name = 'a')
b = tf.constant(5,name = 'b')
c = tf.add(a,b)
print(c.graph)
# 查看默認的計算圖中的Operators
tf.get_default_graph().get_operations()
執行后輸出為:
推薦使用示范1的方式定義計算圖,不用每次都 tf.reset_default_graph()。
上述內容就是TensorFlow基礎中怎么創建計算圖,你們學到知識或技能了嗎?如果還想學到更多技能或者豐富自己的知識儲備,歡迎關注億速云行業資訊頻道。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。