近日在使用tensorflow的过程中,对于tf.assign函数使用有点迷惑,经过仔细看API才发现这函数的妙处。
tf.assign(ref, value, validate_shape=None, use_locking=None, name=None)
函数完成了将value赋值给ref的作用。其中:ref 必须是tf.Variable创建的tensor,如果ref=tf.constant()会报错!
同时,shape(value)==shape(ref)
example:
import tensorflow as tf;
A = tf.Variable([1,2,3])
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print (sess.run(A) ) ## [1,2,3]
sess.run(tf.assign(A, [1,2,5]))
print (sess.run(A) ) ##输出 [1,2,5]