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

operacja

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
)