flux tensoriel : : opérations : : SparseAccumulatorTakeGradient

Extrait le gradient clairsemé moyen dans un SparseConditionalAccumulator .

Résumé

L'opération se bloquera 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 renverra sa 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 SparseConditionalAccumulator .
• 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 :

• Indices Output : Indices de la moyenne des gradients clairsemés accumulés.
• Valeurs Output : valeurs de la moyenne des gradients clairsemés accumulés.
• Forme Output : forme de la moyenne des gradients clairsemés accumulés.

Attributs publics

::tensorflow::Output indices

Operation operation

::tensorflow::Output shape

::tensorflow::Output values

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