TensorFlow-tf.assign()函数解析

tf.assign(A, new_number): 这个函数的功能主要是把A的值变为new_number

#
import tensorflow as tf

#创建一个变量，初始化为标量0
state=tf.Variable(0,name='counter')

#创建一个op，其作用是使state增加1
one=tf.constant(1)
new_value=tf.add(state,one)
update=tf.assign(state,new_value)#tf.assign这个函数的功能主要是把state的值变为new_value


#启动图后，变量必须先经过‘初始化’（init) op初始化
#首先必须增加一个‘初始化’op到图中
init_op=tf.global_variables_initializer()

#启动图，运行op

with tf.Session() as sess:
    #运行‘init' op
    sess.run(init_op)
    #打印“state”的初始值
    print (sess.run(state))
    #运行op，更新’state',并打印‘state’
    for _ in range(3):
        sess.run(update)
        print (sess.run(state))

运行结果：

 

欢迎扫码关注我的微信公众号