Extrait le gradient moyen dans le ConditionalAccumulator donné.
L'opération se bloque 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 renvoie la 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.
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| Sortie <T> | moyenne () La moyenne des gradients accumulés. |
| statique <T> ResourceAccumulatorTakeGradient <T> |
Méthodes héritées
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique d'un tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static ResourceAccumulatorTakeGradient <T> create ( Scope scope, Operand <?> handle, Operand <Integer> numRequired, Class<T> dtype)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ResourceAccumulatorTakeGradient.
Paramètres
| portée | portée actuelle |
|---|---|
| poignée | La poignée d'un accumulateur. |
| numObligatoire | Nombre de dégradés requis avant de renvoyer un agrégat. |
| type | Le type de données des gradients accumulés. Doit correspondre au type d'accumulateur. |
Retour
- une nouvelle instance de ResourceAccumulatorTakeGradient