Spec-Zone.ru › TensorFlow 2.3

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

Spec-Zone.ru

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