assign函数

83 阅读 0 评论 55 点赞

我是靠谱客的博主漂亮花生，这篇文章主要介绍assign函数，现在分享给大家，希望可以做个参考。

tf.assign(A, new_number)的功能是把A的值变为new_number：

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import tensorflow as tf

state = tf.Variable(0, name='counter')
one = tf.constant(1)  # 定义常量one
new_value = tf.add(state, one)  # 定义加法步骤(注意，此步并没有直接计算)
update = tf.assign(state, new_value)  # 将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))

assign函数也可以用于给图变量赋值：

复制代码

1
2
3
4
5
6
7
8
9
10
In [1]: import tensorflow as tf
In [2]: v = tf.Variable(3, name='v')
In [3]: v2 = v.assign(5)
In [4]: sess = tf.InteractiveSession()
In [5]: sess.run(v.initializer)
In [6]: sess.run(v)
Out[6]: 3
In [7]: sess.run(v2)
Out[7]: 5