Variable是一種特殊的數據,它在途中有固定的位置,不像普通張量那樣可以流動,創建變量張量,使用tf.Variable()構造函數,這個構造函數需要一個初始值,初始值的形狀和類型決定了這個變量的形狀和類型。
如:state = tf.Variable(0,name = "counter") 創建一個變量,初始化爲標量0
一段性感的代碼:
import tensorflow as tf
state = tf.Variable(0,name = 'counter')
#常量
one = tf.constant(1)
#定義加法步驟(只是定義,所有運算要到session纔開始)
new_value = tf.add(state,one)
#將State更新成new_value
update = tf.assign(state,new_value)
#如果定義了Variable,就一定要initialize
init = tf.global_variables_initializer()
#使用Session()
with tf.Session() as sess:
sess.run(init)
for i in range(6):
sess.run(update)
print(sess.run(state))