tensorflow :: opérations :: SparseAccumulatorTakeGradient
#include <data_flow_ops.h>Extrait le gradient clairsemé moyen dans un SparseConditionalAccumulator .
Résumé
L'op se 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 retournera 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.
Retour:
- Indices de
Output: Indices de la moyenne des gradients clairsemés accumulés. - Valeurs de
Output: valeurs de la moyenne des gradients clairsemés accumulés. - Forme de
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
façonner
::tensorflow::Output shape
valeurs
::tensorflow::Output values
Fonctions publiques
SparseAccumulatorTakeGradient
SparseAccumulatorTakeGradient( const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_required, DataType dtype )