1. 程式人生 > >tensorflow函式-tf.assign()

tensorflow函式-tf.assign()

下面記錄一個很簡單的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來顯示一個圖模型.