тензорный поток:: опс:: РазреженныйАккумулятор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 )