tensorflow::ops::SparseAccumulatorTakeGradient

#include <data_flow_ops.h>

Extrait le gradient clairsemé moyen dans un SparseConditionalAccumulator .

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
  • handle : Le handle d'un SparseConditionalAccumulator .
  • 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 :

  • Indices Output : Indices de la moyenne des gradients épars accumulés.
  • Valeurs Output : Valeurs de la moyenne des gradients épars accumulés.
  • Forme Output : Forme de la moyenne des gradients épars accumulés.

Constructeurs et destructeurs

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

attributs publics

indices
operation
shape
values

attributs publics

indices

::tensorflow::Output indices

opération

Operation operation

forme

::tensorflow::Output shape

valeurs

::tensorflow::Output values

fonctions publiques

Accumulateur clairsemé Prendre le gradient

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