tf.assign,tf.assign_add,tf.assign_sub

Posted wzdly

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tf.assign,tf.assign_add,tf.assign_sub相关的知识,希望对你有一定的参考价值。

a = tf.Variable(0.0,dtype=tf.float32)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(a))
    a = tf.assign(a,10)
    print(sess.run(a))
    a = tf.assign(a,20)
    print(sess.run(a))
0.0
10.0
20.0


技术分享图片

技术分享图片

a = tf.Variable(1,dtype=tf.float32)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(a))
    a = tf.assign_add(a,10)
    print(sess.run(a))
    a = tf.assign_sub(a,20)
    print(sess.run(a))
1.0
11.0
1.0

 



以上是关于tf.assign,tf.assign_add,tf.assign_sub的主要内容,如果未能解决你的问题,请参考以下文章

吴裕雄 python深度学习与实践(12)

TensorFlow 辨异 —— tf.add(a, b) 与 a+b(tf.assign 与 =)tf.nn.bias_add 与 tf.add(转)

TensorFlow 变量Variable

莫烦TENSORFLOW-Variable

CS 20_Overview of Tensorflow

如何在 TensorFlow 中执行 PyTorch 风格的张量切片更新?