XlaSparseCoreAdam

classe finale publique XlaSparseCoreAdam

Méthodes publiques

statique XlaSparseCoreAdam
créer ( Portée de portée , Opérande <Float> embeddingTable, Opérande <Integer> indices, Opérande <Float> gradient, Opérande <Float> learningRate, Opérande <Float> élan, Opérande <Float> vitesse, Opérande <Float> beta1, Opérande <Float> > beta2, opérande <Float> epsilon, Long featureWidth, booléen useSumInsideSqrt)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaSparseCoreAdam.
Sortie <Flottant>
Sortie <Flottant>
Sortie <Flottant>

Méthodes héritées

Méthodes publiques

public static XlaSparseCoreAdam create ( Scope scope, Operand <Float> embeddingTable, Operand <Integer> indices, Operand <Float> gradient, Operand <Float> learningRate, Operand <Float> momentum, Operand <Float> vitesse, Operand <Float> beta1, Opérande <Float> beta2, Opérande <Float> epsilon, Long featureWidth, Boolean useSumInsideSqrt)

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

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de XlaSparseCoreAdam

Sortie publique <Float> mise à jourEmbeddingTable ()

Sortie publique <Float> mise à jourMomentum ()

Sortie publique <Float> mise à jourVelocity ()