flux tensoriel : : opérations : : SparseAccumulatorTakeGradient

#include <data_flow_ops.h>

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.

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

SparseAccumulatorTakeGradient 

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