assign函数

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

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函数也可以用于给图变量赋值：

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

最后

以上就是漂亮花生为你收集整理的assign函数的全部内容，希望文章能够帮你解决assign函数所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。