tf.assign(ref, value, validate_shape=None, use_locking=None, name=None),
函数功能是将value赋值给ref
ref必须是tf.Variable创建的tensor,如果ref=tf.constant()就会报错,而且默认情况下ref的shape和value的shape是相同的
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.global_variables_initializer() with tf.Session() as sess: sess.run(init) for __ in range(3): sess.run(update) print(sess.run(state)) #counter:0 1 2 3