Tensorflow從入門到禿頂(2)

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))

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章