tensorflow::ops::AccumulatorTakeGradient

#include <data_flow_ops.h>

Extrait le gradient moyen dans l' accumulateur conditionnel donné.

Résumé

L'opération se bloque jusqu'à ce qu'un nombre suffisant de gradients (c'est-à-dire supérieur à `num_required`) ait été accumulé. Si l'accumulateur a déjà agrégé plus de `num_required` gradients, il renvoie la moyenne des gradients accumulés. Il incrémente également automatiquement de 1 la valeur de `global_step` enregistrée dans l'accumulateur et réinitialise l'agrégat à 0.

Arguments :

  • portée : un objet Scope
  • poignée : La poignée d'un accumulateur.
  • num_required : Nombre de gradients requis avant de renvoyer un agrégat.
  • dtype : Type de données des gradients accumulés. Doit correspondre au type de l’accumulateur.

Retours :

  • Output : La moyenne des gradients cumulés.

Constructeurs et destructeurs

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

attributs publics

average
operation

fonctions publiques

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

attributs publics

moyenne

::tensorflow::Output average

opération

Operation operation

fonctions publiques

AccumulateurPrendreGradient

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

nœud

::tensorflow::Node * node() const 

opérateur::tensorflow::Entrée

 operator::tensorflow::Input() const 

opérateur::tensorflow::Sortie

 operator::tensorflow::Output() const