przepływ tensorowy:: ops:: Akumulator WeźGradient

#include <data_flow_ops.h>

Wyodrębnia średni gradient w danym ConditionalAccumulator .

Streszczenie

Op blokuje się 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, zwraca ś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
  • uchwyt: Uchwyt do akumulatora.
  • num_required: Liczba gradientów wymaganych, zanim zwrócimy agregację.
  • dtype: Typ danych skumulowanych gradientów. Musi odpowiadać typowi akumulatora.

Zwroty:

  • Output : Średnia zgromadzonych gradientów.

Konstruktory i destruktory

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

Atrybuty publiczne

average
operation

Funkcje publiczne

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atrybuty publiczne

przeciętny 

::tensorflow::Output average

działanie 

Operation operation

Funkcje publiczne

Akumulator WeźGradient 

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

węzeł 

::tensorflow::Node * node() const

operator::tensorflow::Wejście 

 operator::tensorflow::Input() const

operator::tensorflow::Wyjście 

 operator::tensorflow::Output() const