XlaSparseCoreAdam

classe final pública XlaSparseCoreAdam

Métodos Públicos

XlaSparseCoreAdam estático
create ( Escopo do escopo , Operando <Float> embeddingTable, Operando <Integer> índices, Operando <Float> gradiente, Operando <Float> learningRate, Operando <Float> momentum, Operando <Float> velocidade, Operando <Float> beta1, Operando <Float> > beta2, Operando <Float> épsilon, Long featureWidth, Boolean useSumInsideSqrt)
Método de fábrica para criar uma classe que envolve uma nova operação XlaSparseCoreAdam.
Saída <flutuante>
Saída <flutuante>
Saída <flutuante>

Métodos herdados

Métodos Públicos

public static XlaSparseCoreAdam create ( Escopo de escopo , Operando <Float> embeddingTable, Operando <Integer> índices, Operando <Float> gradiente, Operando <Float> learningRate, Operando <Float> momentum, Operando <Float> velocidade, Operando <Float> beta1, Operando <Float> beta2, Operando <Float> épsilon, Long featureWidth, Boolean useSumInsideSqrt)

Método de fábrica para criar uma classe que envolve uma nova operação XlaSparseCoreAdam.

Parâmetros
escopo escopo atual
Devoluções
  • uma nova instância de XlaSparseCoreAdam

Saída pública <Float> atualizadaEmbeddingTable ()

Saída pública <Float> atualizadoMomentum ()

Saída pública <Float> atualizadaVelocity ()