TensorFlow的assign赋值用法

最新推荐文章于 2024-07-18 08:46:24 发布

mustar_2017

最新推荐文章于 2024-07-18 08:46:24 发布

阅读量2.1w

点赞数 3

本文链接：https://blog.csdn.net/mustar_2017/article/details/79336679

版权

在TensorFlow中，修改变量值后需要通过assign操作来更新。常见错误用法及三种正确的赋值方法，包括直接使用assign，assign_add和assign_sub。详细解答见StackOverflow链接。

摘要由CSDN通过智能技术生成

TensorFlow修改变量值后，需要重新赋值，assign用起来有点小技巧，就是需要需要弄个操作子，运行一下。

下面这么用是不行的

import tensorflow as tf
import numpy as np

x = tf.Variable(0)
init = tf.initialize_all_variables()
sess = tf.InteractiveSession()
sess.run(init)

print(x.eval())

x.assign(1)
print(x.eval())

正确用法

https://stackoverflow.com/questions/34220532/how-to-assign-a-value-to-a-tensorflow-variable提供了3种方法

import tensorflow as tf
x = tf.Variable(0)
y = tf.assign(x, 1)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print sess.run(x)
    print sess.run(y)
    print sess.run(x)

In [212]: w = tf.Variable(12)
In [213]: w_new = w.assign(34)

In [214]: with tf.Session() as sess:
     ...:     sess.run(w_new)
     ...:     print(w_new.eval())

# output
34