ResourceApplyKerasMomentum

classe finale publique ResourceApplyKerasMomentum

Mettez à jour '*var' selon le schéma momentum.

Définissez use_nesterov = True si vous souhaitez utiliser l'élan Nesterov.

accumulation = accumulation * élan - lr * grad var += accumulation

Classes imbriquées

classe ResourceApplyKerasMomentum.Options Attributs facultatifs pour ResourceApplyKerasMomentum

Méthodes publiques

statique <T> ResourceApplyKerasMomentum
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 ResourceApplyKerasMomentum.
statique ResourceApplyKerasMomentum.Options
useLocking (booléen useLocking)
statique ResourceApplyKerasMomentum.Options
useNesterov (booléen useNesterov)

Méthodes héritées

Méthodes publiques

public static ResourceApplyKerasMomentum 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 ResourceApplyKerasMomentum.

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 ResourceApplyKerasMomentum

public statique ResourceApplyKerasMomentum.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 ResourceApplyKerasMomentum.Options useNesterov (booléen useNesterov)

Paramètres
utiliser Nesterov Si « Vrai », le tenseur passé au calcul de la graduation sera var + momentum * accum, donc à la fin, la var que vous obtenez est en fait var + momentum * accum.