变量(Variable)的概念
在TensorFlow中,变量(Variable)是特殊的张量(Tensor),它的值可以是一个任何类型和形状的张量。与其他张量不同,变量存在于单个 session.run 调用的上下文之外,也就是说,变量存储的是持久张量,当训练模型时,用变量来存储和更新参数。
- 注意
在调用op之前,所有变量都应被显式地初始化才行。
init = tf.global_variables_initializer()#全局变量初始化
代码
import tensorflow as tf
import os ##忽略AVX扩展警告
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
x = tf.Variable([1, 2])
a = tf.constant([3, 3])
sub = tf.subtract(x, a) # 增加一个减法op
add = tf.add(x, sub) # 增加一个加法op
# 注意变量再使用之前要再sess中做初始化,但是下边这种初始化方法不会指定变量的初始化顺序
init = tf.global_variables_initializer()#全局变量初始化
with tf.Session() as sess:
sess.run(init)
print(sess.run(sub))
print(sess.run(add))
# 创建一个名字为‘counter’的变量 初始化0
state = tf.Variable(0, name='counter')
new_value = tf.add(state, 1) # 创建一个op,作用是使state加1
update = tf.assign(state, new_value) # 赋值op,不能直接用等号赋值,作用是 state = new_value,借助tf.assign()函数
init = tf.global_variables_initializer()#全局变量初始化
with tf.Session() as sess:
sess.run(init)
print(sess.run(state))
for _ in range(5):#循环5次
sess.run(update)
print(sess.run(state))