tensorflow函式-tf.assign()
阿新 • • 發佈:2019-01-24
下面記錄一個很簡單的tensorflow程式碼,用來理解tensorflow執行機制
程式碼中有詳細註釋,大家可以在執行一下,為什麼會是1,2,3這個結果
#-*-coding:UTF-8-*-
#這句話是指定*.py的編碼方式,如果檔案中涉及到中文漢字的話,有必要寫一下這句話。當然也可以這樣寫:encoding:UTF-8
import tensorflow as tf
state=tf.Variable(0,name="a")#使用tensorflow在預設的圖中建立節點,這個節點是一個變數
one = tf.constant(1)#此處呼叫了tf的一個函式,用於建立常量
new_value = tf.add (state,one)#對常量與變數進行簡單的加法操作,
#這點需要說明的是: 在TensoorFlow中,所有的操作op,變數都視為節點,tf.add()
#的意思就是在tf的預設圖中新增一個op,這個op是用來做加法操作的。
update = tf.assign(state,new_value)#這個操作是:賦值操作。將new_value的值賦值給state變數,update只是一個用於sess
#的變數
init=tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init)#對變數進行初始化,執行(run)init語句
for i in range(3):
sess.run(update)
print(sess.run(state))
下圖是執行結果
tensorflow的圖中tensor的流動模型如下
後面我會用tensorboard來顯示一個圖模型.