Метод assign класса Variable

Метод assign класса Variable присваивает переменной новое значение. Первым параметром метод принимает новое значение, которое может быть числом, списком, массивом или тензором. Вторым параметром можно передать флаг use_locking, который определяет, нужно ли использовать блокировку при присваивании. Метод возвращает тензор с новым значением переменной.

Синтаксис

variable.assign(value, [use_locking])

Пример

Давайте создадим переменную и присвоим ей новое значение:

import tensorflow as tf

v = tf.Variable([1, 2, 3, 4, 5])
res = v.assign([6, 7, 8, 9, 10])
print(v)

Результат выполнения кода:

<tf.Variable 'Variable:0' shape=(5,) dtype=int32, numpy=array([ 6, 7, 8, 9, 10], dtype=int32)>

Пример

Давайте присвоим переменной новое значение из тензора:

import tensorflow as tf

v = tf.Variable([1, 2, 3, 4, 5])
t = tf.constant([10, 20, 30, 40, 50])
res = v.assign(t)
print(v)

Результат выполнения кода:

<tf.Variable 'Variable:0' shape=(5,) dtype=int32, numpy=array([10, 20, 30, 40, 50], dtype=int32)>

Пример

Давайте присвоим переменной новое значение с блокировкой:

import tensorflow as tf

v = tf.Variable([1, 2, 3, 4, 5])
res = v.assign([7, 8, 9, 10, 11], use_locking=True)
print(v)

Результат выполнения кода:

<tf.Variable 'Variable:0' shape=(5,) dtype=int32, numpy=array([ 7, 8, 9, 10, 11], dtype=int32)>

Смотрите также

класс Variable,
который создает переменную TensorFlow
метод assign_add,
который прибавляет значение к переменной
метод assign_sub,
который вычитает значение из переменной
метод read_value,
который возвращает значение переменной