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 )