tf.compat.v1.ConditionalAccumulator
Условный аккумулирующий объект для агрегирования градиентов.
Наследуется от: ConditionalAccumulatorBase
tf.compat.v1.ConditionalAccumulator(
dtype,
shape=None,
shared_name=None,
name='conditional_accumulator',
reduction_type='MEAN'
)
Актуальные градиенты (т.е. временная метка, в которой был вычислен градиент, равна временной метке аккумулирующего объекта) добавляются в аккумулирующий объект.
Извлечение среднего градиента блокируется до тех пор, пока не будет накоплено необходимое количество градиентов.
| Аргументы | |
|---|---|
dtype | Тип данных накапливаемых градиентов. |
shape | Форма накапливаемых градиентов. |
shared_name | Необязательно. Если не пусто, этот аккумулирующий объект будет совместно использоваться под заданным именем в нескольких сессиях. |
name | Необязательное имя для аккумулирующего объекта. |
reduction_type | Тип редукции для использования при вычислении градиента. |
| Атрибуты | |
|---|---|
accumulator_ref | Ссылка на базовый аккумулирующий объект. |
dtype | Тип данных градиентов, аккумулированных этим аккумулирующим объектом. |
name | Имя базового аккумулирующего объекта. |
Методы
apply_grad
apply_grad(
grad, local_step=0, name=None
)
Попытка применить градиент к аккумулирующему объекту.
Попытка будет проигнорирована, если градиент устарел, т.е. local_step меньше глобальной временной метки аккумулирующего объекта.
| Аргументы | |
|---|---|
grad | Тензор градиента, который необходимо применить. |
local_step | Временная метка, в которой был вычислен градиент. |
name | Необязательное имя для операции. |
| Возвращаемые значения | |
|---|---|
| Операция, которая (условно) применяет градиент к аккумулирующему объекту. |
| Исключения | |
|---|---|
ValueError | Если форма grad неверна |
num_accumulated
num_accumulated(
name=None
)
Количество градиентов, которые в настоящее время были агрегированы в аккумулирующем объекте.
| Аргументы | |
|---|---|
name | Необязательное имя для операции. |
| Возвращаемые значения | |
|---|---|
| Количество накопленных градиентов в данный момент в аккумулирующем объекте. |
set_global_step
set_global_step(
new_global_step, name=None
)
Устанавливает глобальную временную метку аккумулирующего объекта.
Операция выводит предупреждение, если мы пытаемся установить временную метку, меньшую, чем временная метка самого аккумулирующего объекта.
| Аргументы | |
|---|---|
new_global_step | Значение новой временной метки. Может быть переменной или константой. |
name | Необязательное имя для операции. |
| Возвращаемые значения | |
|---|---|
| Операция, устанавливающая временную метку аккумулирующего объекта. |
take_grad
take_grad(
num_required, name=None
)
Попытка извлечь средний градиент из аккумулирующего объекта.
Операция блокируется, пока достаточное количество градиентов успешно не будет применено к аккумулирующему объекту.
При успешном выполнении также выполняются следующие действия:
- Счетчик накопленных градиентов сбрасывается до 0.
- Агрегированный градиент сбрасывается до нулевого тензора.
- Внутренняя временная метка аккумулирующего объекта увеличивается на 1.
| Аргументы | |
|---|---|
num_required | Количество градиентов, которые должны быть агрегированы. |
name | Необязательное имя для операции. |
| Возвращаемые значения | |
|---|---|
| Тензор, содержащий значение среднего градиента. |
| Исключения | |
|---|---|
InvalidArgumentError | Если num_required < 1 |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/ConditionalAccumulator