溫馨提示×

tensorflow中assign的用法是什么

小億
118
2024-04-11 12:25:09

在TensorFlow中,assign操作用于將一個值分配給一個TensorFlow變量。它可以用來更新變量的值。

例如,假設(shè)我們有一個變量x,我們想要將其值設(shè)置為5,可以使用assign操作如下:

import tensorflow as tf

x = tf.Variable(0, dtype=tf.int32)
assign_op = x.assign(5)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    sess.run(assign_op)
    print(sess.run(x))  # 輸出為5

在這個例子中,我們首先創(chuàng)建一個變量x,然后使用assign操作將其值設(shè)置為5。最后,我們在會話中運行assign操作,并輸出變量x的值。

0