ResourceApplyMomentum

classe finale publique ResourceApplyMomentum

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 ResourceApplyMomentum.Options Attributs facultatifs pour ResourceApplyMomentum

Constantes

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

Méthodes publiques

statique <T étend TType > ResourceApplyMomentum
create ( Scope scope, Opérande <?> var, Opérande <?> accum, Opérande <T> lr, Opérande <T> grad, Opérande <T> momentum, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ResourceApplyMomentum.
statique ResourceApplyMomentum.Options
useLocking (booléen useLocking)
statique ResourceApplyMomentum.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

Valeur constante : "ResourceApplyMomentum"

Méthodes publiques

public static ResourceApplyMomentum create ( Scope scope, Operand <?> var, Operand <?> 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 ResourceApplyMomentum.

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 ResourceApplyMomentum

public statique ResourceApplyMomentum.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 ResourceApplyMomentum.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.