tensorflow:: ops:: AccumulatorTakeGradient

#include <data_flow_ops.h>

Wyodrębnia średni gradient w podanym obiekcie ConditionalAccumulator .

Streszczenie

Operacja blokuje się, aż do momentu zgromadzenia wystarczającej liczby gradientów (tj. większej niż num_required). Jeśli akumulator zgromadził już więcej niż num_required gradientów, zwraca średnią z zakumulowanych gradientów. Automatycznie zwiększa również zarejestrowany global_step w akumulatorze o 1 i resetuje agregat do 0.

Argumenty:

  • zakres: obiekt zakresu
  • uchwyt: Uchwyt akumulatora.
  • num_required: Liczba gradientów wymaganych przed zwróceniem agregatu.
  • dtype: Typ danych skumulowanych gradientów. Musi odpowiadać typowi akumulatora.

Zwroty:

  • Output : Średnia skumulowanych 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

AkumulatorTakeGradient

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

węzeł

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const