Mettez à jour '*var' selon le schéma momentum.
Définissez use_nesterov = True si vous souhaitez utiliser l'élan Nesterov.
accumulation = accumulation * élan + grad var -= lr * accumulation
Classes imbriquées
classe | AppliquerMomentum.Options | Attributs facultatifs pour ApplyMomentum |
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. |
statique <T étend TType > ApplyMomentum <T> | |
Sortie <T> | dehors () Identique à "var". |
statique ApplyMomentum.Options | useLocking (booléen useLocking) |
statique ApplyMomentum.Options | useNesterov (booléen useNesterov) |
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
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 static ApplyMomentum <T> create ( Scope scope, Operand <T> var, Operand <T> accum, Operand <T> lr, Operand <T> grad, Operand <T> momentum, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ApplyMomentum.
Paramètres
portée | portée actuelle |
---|---|
var | Doit provenir d'une variable (). |
cumuler | Doit provenir d'une variable (). |
g / D | Facteur d'échelle. Ça doit être un scalaire. |
diplômé | Le dégradé. |
élan | Élan. Ça doit être un scalaire. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de ApplyMomentum
public static ApplyMomentum.Options useLocking (booléen useLocking)
Paramètres
utiliserVerrouillage | Si `True`, la mise à jour des tenseurs var et cumul sera protégée par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits. |
---|
public static ApplyMomentum.Options useNesterov (booléen useNesterov)
Paramètres
utiliser Nesterov | Si « Vrai », le tenseur passé au calcul de grad sera var - lr * momentum * accum, donc à la fin, la var que vous obtenez est en fait var - lr * momentum * accum. |
---|