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.
- Агрегированный градиент сбрасывается до 0 тензора.
- Внутренний шаг времени аккумулятора увеличивается на 1.
| Аргументы | |
|---|---|
num_required | Количество градиентов, которые необходимо было агрегировать |
name | Необязательное имя для операции |
| Возвращаемое значение | |
|---|---|
| Тензор, содержащий значение среднего градиента. |
| Исключения | |
|---|---|
InvalidArgumentError | Если num_required < 1 |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/ConditionalAccumulator