tf.Variable

一個變量通過調(diào)用run() 方法維持圖的狀態(tài)。你通過構(gòu)造variable 類的實例來添加一個變量到圖中。

Variable() 構(gòu)造器需要一個初始值，可以是任意類型和shape 的Tensor。初始值定義了變量的type和shape。構(gòu)造完成之后，變量的type和shape 是固定的。可以使用assign 方法來修改變量的值。

如果你想修改變量的shape，你必須使用assign 操作，并且 validate_shpe=False

就像任何Tensor，通過Variable() 創(chuàng)建的variable，可以用作圖中其他操作節(jié)點的輸入。另外，所有操作承載的Tensor 類傳遞給variables. 所以你可以僅僅通過對變量執(zhí)行算術(shù)來對圖中添加節(jié)點。

代碼：

import tensorflow as tf

state = tf.Variable(0, name='counter')
print(state.name)
one = tf.constant(1)


new_value = tf.add(state, one)
update = tf.assign(state, new_value)

#init = tf.initialize_all_variables()? # deprecated
init = tf.global_variables_initializer()


with tf.Session() as sess:
??? sess.run(init)
??? for _ in range(3):
??????? sess.run(update)
??????? print(sess.run(state))

結(jié)果：

counter_2:0
1
2
3

總結(jié)

以上是生活随笔為你收集整理的TensorFlow学习笔记（六）Variable变量的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。