Spec-Zone.ru › TensorFlow 2.9

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API