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

operacja

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