XlaSparseCoreAdam

classe finale pubblica XlaSparseCoreAdam

Metodi pubblici

XlaSparseCoreAdam statico
create ( Scope scope, Operando <Float> embeddingTable, Indici operando <Integer>, Operando <Float> gradiente, Operando <Float> learningRate, Operando <Float> momentum, Operando <Float> velocità, Operando <Float> beta1, Operando <Float > beta2, operando <Float> epsilon, featureWidth lunga, uso booleano SumInsideSqrt)
Metodo factory per creare una classe che racchiude una nuova operazione XlaSparseCoreAdam.
Uscita <Mobile>
Uscita <Mobile>
Uscita <Mobile>

Metodi ereditati

Metodi pubblici

public static XlaSparseCoreAdam create ( Scope scope, Operando <Float> embeddingTable, Operando <Integer> indici, Operando <Float> gradiente, Operando <Float> learningRate, Operando <Float> momentum, Operando <Float> velocità, Operando <Float> beta1, Operando <Float> beta2, Operando <Float> epsilon, Long featureWidth, Boolean useSumInsideSqrt)

Metodo factory per creare una classe che racchiude una nuova operazione XlaSparseCoreAdam.

Parametri
scopo ambito attuale
ritorna
  • una nuova istanza di XlaSparseCoreAdam

output pubblico <Float> aggiornatoEmbeddingTable ()

Output pubblico <Float> aggiornatoMomentum ()

output pubblico <Float> aggiornatoVelocity ()