tf.compat.v1.assign_add

Обновляет ref, добавляя к нему value.

tf.compat.v1.assign_add(
    ref, value, use_locking=None, name=None
)

Перейдите на TF2

tf.compat.v1.assign_add в основном совместим с eager execution и tf.function.

Чтобы перейти на нативный стиль TF2, можно использовать метод 'assign_add' объекта tf.Variable:

Как сопоставить аргументы

Имя аргумента TF1	Имя аргумента TF2	Примечание
`ref`	`self`	В методе `assign_add()`
`value`	`value`	В методе `assign_add()`
`use_locking`	`use_locking`	В методе `assign_add()`
`name`	`name`	В методе `assign_add()`
-	`read_value`	Установите значение True, чтобы воспроизвести поведение (по умолчанию — True)

Пример использования до и после

До:

with tf.Graph().as_default():
  with tf.compat.v1.Session() as sess:
    a = tf.compat.v1.Variable(0, dtype=tf.int64)
    sess.run(a.initializer)
    update_op = tf.compat.v1.assign_add(a, 1)
    res_a = sess.run(update_op)
    res_a
1

После:

b = tf.Variable(0, dtype=tf.int64)
res_b = b.assign_add(1)
res_b.numpy()
1

Описание

Эта операция возвращает ref после завершения обновления. Это упрощает объединение в цепочку операций, которым требуется использовать сброшенное значение. В отличие от tf.math.add, эта операция не выполняет широковещательную рассылку. ref и value должны иметь одинаковую форму.

Аргументы
`ref`	Изменяемый объект `Tensor`. Должен иметь один из следующих типов: `float32`, `float64`, `int64`, `int32`, `uint8`, `uint16`, `int16`, `int8`, `complex64`, `complex128`, `qint8`, `quint8`, `qint32`, `half`. Должен быть получен из узла `Variable`.
`value`	Объект `Tensor`. Должен иметь ту же форму и тип данных, что и `ref`. Значение, добавляемое к переменной.
`use_locking`	Необязательный объект `bool`. По умолчанию — `False`. Если значение True, сложение будет защищено блокировкой; в противном случае поведение не определено, но может сопровождаться меньшей конкуренцией за ресурсы.
`name`	Имя операции (необязательно).

Возвращает
То же, что и `ref`. Возвращается для удобства операций, которым требуется использовать новое значение после обновления переменной.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/assign_add