tf.compat.v1.SparseConditionalAccumulator

Условный аккумулирующий агрегат для агрегирования разреженных градиентов.

Наследуется от: ConditionalAccumulatorBase

tf.compat.v1.SparseConditionalAccumulator(
    dtype,
    shape=None,
    shared_name=None,
    name='sparse_conditional_accumulator',
    reduction_type='MEAN'
)

Разреженные градиенты представлены как IndexedSlices.

Актуальные градиенты (то есть шаг времени, в котором был вычислен градиент, равен шагу времени агрегатора) добавляются в агрегатор.

Извлечение среднего градиента блокируется до тех пор, пока не будет накоплено необходимое количество градиентов.

Методы

apply_grad

Просмотреть исходный код

apply_grad(
    grad_indices, grad_values, grad_shape=None, local_step=0, name=None
)

Попытка применить разреженный градиент к агрегатору.

Попытка будет проигнорирована, если градиент устарел, т. е., local_step меньше шага времени агрегатора.

Разреженный градиент представлен своими индексами, значениями и, возможно, пустой или None формой. Индексы должны быть вектором, представляющим расположение ненулевых элементов в тензоре. Значения — ненулевые срезы градиента, и должны иметь такое же первое измерение, как индексы, т. е., nnz, представленные индексами и значениями, должны быть согласованы. Форма, если она не пустая или None, должна быть согласована с формой агрегатора (если она также задана).

apply_indexed_slices_grad

Просмотреть исходный код

apply_indexed_slices_grad(
    grad, local_step=0, name=None
)

Попытка применить градиент к агрегатору.

Попытка будет проигнорирована, если градиент устарел, т. е., local_step меньше шага времени агрегатора.

num_accumulated

Просмотреть исходный код

num_accumulated(
    name=None
)

Количество градиентов, которые в настоящее время агрегированы в агрегаторе.

set_global_step

Просмотреть исходный код

set_global_step(
    new_global_step, name=None
)

Устанавливает глобальный шаг времени агрегатора.

Операция записывает предупреждение, если мы пытаемся установить шаг времени, который меньше текущего шага времени агрегатора.

take_grad

Просмотреть исходный код

take_grad(
    num_required, name=None
)

Попытка извлечь средний градиент из агрегатора.

Операция блокируется до тех пор, пока достаточное количество градиентов не будет успешно применено к агрегатору.

После успешного выполнения также выполняются следующие действия:

• Счетчик накопленных градиентов сбрасывается до 0.
• Агрегированный градиент сбрасывается до нулевого тензора.
• Внутренний шаг времени агрегатора увеличивается на 1.

take_indexed_slices_grad

Просмотреть исходный код

take_indexed_slices_grad(
    num_required, name=None
)

Попытка извлечь средний градиент из агрегатора.

Операция блокируется до тех пор, пока достаточное количество градиентов не будет успешно применено к агрегатору.

После успешного выполнения также выполняются следующие действия:

• Счетчик накопленных градиентов сбрасывается до 0.
• Агрегированный градиент сбрасывается до нулевого тензора.
• Внутренний шаг времени агрегатора увеличивается на 1.