flux tensoriel : : opérations : : AccumulateurTakeGradient

#include <data_flow_ops.h>

Extrait le gradient moyen dans le ConditionalAccumulator donné.

Résumé

L'opération se bloque jusqu'à ce que suffisamment de gradients (c'est-à-dire plus que num_required) aient été accumulés. Si l'accumulateur a déjà agrégé plus de num_required gradients, il renvoie la moyenne des gradients accumulés. Incrémente également automatiquement le global_step enregistré dans l'accumulateur de 1 et réinitialise l'agrégat à 0.

Arguments :

  • scope : un objet Scope
  • handle : le handle d'un accumulateur.
  • num_required : nombre de dégradés requis avant de renvoyer un agrégat.
  • dtype : le type de données des dégradés accumulés. Doit correspondre au type d'accumulateur.

Retours :

  • Output : La moyenne des gradients accumulé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

AccumulateurTakeGradient

 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