Tensorflow—变量

代码:

import tensorflow as tf 

x = tf.Variable([1, 2])
a = tf.constant([3, 3])
#增加一个减法op
sub = tf.subtract(x, a)
#增加一个加法op
add = tf.add(x, sub)

#变量要进行初始化
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]

代码:

#变量是可以起名字的哦
#创建一个变量初始化为0
state = tf.Variable(0, name='counter')

#创建一个op,作用是使state加1
new_value = tf.add(state, 1)
#创建赋值op,将new_value赋值给state
updata = tf.assign(state, new_value)

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

with tf.Session() as sess:
    sess.run(init)
    print(sess.run(state))
    #循环执行5次
    for _ in range(5):
        sess.run(updata)
        print(sess.run(state))

运行结果:

0
1
2
3
4
5

猜你喜欢

转载自blog.csdn.net/wangsiji_buaa/article/details/80200706