ResourceSparseApplyKerasMomentum

classe finale publique ResourceSparseApplyKerasMomentum

Mettez à jour les entrées pertinentes dans '*var' et '*accum' selon le schéma momentum.

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

C'est pour les lignes pour lesquelles nous avons grad, nous mettons à jour var et accumulons comme suit :

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

Classes imbriquées

classe ResourceSparseApplyKerasMomentum.Options Attributs facultatifs pour ResourceSparseApplyKerasMomentum

Méthodes publiques

statique <T, U étend le nombre> ResourceSparseApplyKerasMomentum
create ( Scope scope, Opérande <?> var, Opérande <?> accum, Opérande <T> lr, Opérande <T> grad, Opérande <U> indices, Opérande <T> momentum, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ResourceSparseApplyKerasMomentum.
statique ResourceSparseApplyKerasMomentum.Options
useLocking (booléen useLocking)
statique ResourceSparseApplyKerasMomentum.Options
useNesterov (booléen useNesterov)

Méthodes héritées

Méthodes publiques

public static ResourceSparseApplyKerasMomentum create ( Scope scope, Operand <?> var, Operand <?> accum, Operand <T> lr, Operand <T> grad, Operand <U> indices, Operand <T> momentum, Options... options)

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

Paramètres
portée portée actuelle
var Doit provenir d'une variable ().
cumuler Doit provenir d'une variable ().
g / D Taux d'apprentissage. Ça doit être un scalaire.
diplômé Le dégradé.
indices Un vecteur d'indices dans la première dimension de var et cumul.
élan Élan. Ça doit être un scalaire.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de ResourceSparseApplyKerasMomentum

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