тензорный поток:: опс:: РазреженныйАккумуляторTakeGradient

#include <data_flow_ops.h>

Извлекает средний разреженный градиент в SparseConditionalAccumulator .

Краткое содержание

Операция будет блокироваться до тех пор, пока не будет накоплено достаточное количество градиентов (т. е. больше, чем num_required). Если аккумулятор уже собрал более num_required градиентов, он вернет среднее значение накопленных градиентов. Также автоматически увеличивает записанный global_step в аккумуляторе на 1 и сбрасывает агрегат на 0.

Аргументы:

  • область: объект области .
  • handle: Дескриптор SparseConditionalAccumulator .
  • num_required: количество градиентов, необходимое, прежде чем мы вернем агрегат.
  • dtype: тип данных накопленных градиентов. Должен соответствовать типу аккумулятора.

Возврат:

  • Output индексы: индексы среднего значения накопленных разреженных градиентов.
  • Output значения: значения среднего значения накопленных разреженных градиентов.
  • Output форма: форма среднего значения накопленных редких градиентов.

Конструкторы и деструкторы

SparseAccumulatorTakeGradient (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_required, DataType dtype)

Публичные атрибуты

indices
operation
shape
values

Публичные атрибуты

индексы

::tensorflow::Output indices

операция

Operation operation

форма

::tensorflow::Output shape

ценности

::tensorflow::Output values

Общественные функции

РазреженныйАккумуляторTakeGradient

 SparseAccumulatorTakeGradient(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_required,
  DataType dtype
)