AccumulatorTakeGradient

classe finale publique AccumulatorTakeGradient

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.

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique du tenseur.
Sortie <T>
moyenne ()
La moyenne des gradients accumulés.
statique <T étend TType > AccumulatorTakeGradient <T>
créer ( Scope scope, Operand < TString > handle, Operand < TInt32 > numRequired, Class<T> dtype)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération AccumulatorTakeGradient.

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "AccumulatorTakeGradient"

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique du 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 Sortie <T> moyenne ()

La moyenne des gradients accumulés.

public static AccumulatorTakeGradient <T> create ( Scope scope, Operand < TString > handle, Operand < TInt32 > numRequired, Class<T> dtype)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération AccumulatorTakeGradient.

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 AccumulatorTakeGradient