人工智能TensorFlow-变量的使用

2018-10-23  本文已影响3人  东南有大树
import tensorflow as tf 

用Variable()来定义变量

# 定义一个变量
x = tf.Variable([1, 2])
# 定义一个常量 
a = tf.constant([3, 3])

# 定义一个减法的op
sub = tf.subtract(x, a)
# 定义一个加法的op
add = tf.add(x, sub)

# 变量初始化op
init = tf.global_variables_initializer()

# 定义会话
with tf.Session() as sess:
    sess.run(init)  # 初始化变量
    print(sess.run(sub))
    print(sess.run(add))
[-2 -1]
[-1  1]

如果要使用变量,一定要将其初始化

一个循环累加变量值的例子

# 定义一个变量,变量是可以起名字的
state = tf.Variable(0, name='counter')  # 初始化值为0
# 创建一个加法op,让state加1
new_value = tf.add(state, 1)
# 赋值方法,将第二个参数的值赋给第一个参数
update = tf.assign(state, new_value)

# 变量初始化op
init = tf.global_variables_initializer()

# 定义会话
with tf.Session() as sess:
    sess.run(init) 
    print(sess.run(state))
    
    for _ in range(5):
        print(sess.run(update))
0
1
2
3
4
5

op间的联系组成了图,它们需要在Session中运行!

上一篇下一篇

猜你喜欢

热点阅读