przepływ tensorowy:: ops:: SparseAccumulatorTakeGradient
#include <data_flow_ops.h>
Wyodrębnia średni rzadki gradient w SparseConditionalAccumulator .
Streszczenie
Operacja będzie blokować do momentu zgromadzenia wystarczającej liczby (tj. więcej niż num_required) gradientów. Jeśli akumulator zagregował już więcej niż num_required gradientów, zwróci średnią z skumulowanych gradientów. Automatycznie zwiększa również zarejestrowany krok globalny w akumulatorze o 1 i resetuje agregat do 0.
Argumenty:
- zakres: Obiekt Scope
- handle: Uchwyt do SparseConditionalAccumulator .
- num_required: Liczba gradientów wymaganych, zanim zwrócimy agregację.
- dtype: Typ danych skumulowanych gradientów. Musi odpowiadać typowi akumulatora.
Zwroty:
- Wskaźniki
Output
: Wskaźniki średniej skumulowanych rzadkich gradientów. - Wartości
Output
: Wartości średniej zgromadzonych rzadkich gradientów. - Kształt
Output
: Kształt średniej zgromadzonych rzadkich gradientów.
Konstruktory i destruktory | |
---|---|
SparseAccumulatorTakeGradient (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_required, DataType dtype) |
Atrybuty publiczne | |
---|---|
indices | |
operation | |
shape | |
values |
Atrybuty publiczne
indeksy
::tensorflow::Output indices
działanie
Operation operation
kształt
::tensorflow::Output shape
wartości
::tensorflow::Output values
Funkcje publiczne
SparseAccumulatorTakeGradient
SparseAccumulatorTakeGradient( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_required, DataType dtype )