Метод assign_add

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

Синтаксис

var.assign_add(delta, [use_locking])

Пример

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

import tensorflow as tf

v = tf.Variable([1, 2, 3, 4, 5])
v.assign_add(10)
print(v)

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

<tf.Variable 'Variable:0' shape=(5,) dtype=int32, numpy=array([11, 12, 13, 14, 15], dtype=int32)>

Пример

Давайте прибавим к переменной другой тензор той же формы:

<+python+> import tensorflow as tf v = tf.Variable([1, 2, 3, 4, 5]) delta = tf.constant([10, 20, 30, 40, 50]) v.assign_add(delta) print(v) <-python+>

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

<tf.Variable 'Variable:0' shape=(5,) dtype=int32, numpy=array([11, 22, 33, 44, 55], dtype=int32)>

Пример

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

import tensorflow as tf

v = tf.Variable([1.0, 2.0, 3.0], dtype=tf.float32)
v.assign_add(0.5)
print(v)

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

<tf.Variable 'Variable:0' shape=(3,) dtype=float32, numpy=array([1.5, 2.5, 3.5], dtype=float32)>

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

класс Variable,
который создает изменяемый тензор
метод assign,
который присваивает переменной новое значение
метод assign_sub,
который вычитает значение из переменной
метод numpy,
который возвращает значение переменной как массив NumPy