Restez organisé à l'aide des collections Enregistrez et classez les contenus selon vos préférences.

tensorflow :: opérations :: AccumulatorTakeGradient

#include <data_flow_ops.h>

Extrait le gradient moyen dans le ConditionalAccumulator donné.

Résumé

L'op bloque jusqu'à ce que des gradients suffisants (c'est-à-dire plus de 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: La poignée 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.

Retour:

  • 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

AccumulatorTakeGradient

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

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const