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