XlaSparseCoreAdam

clase final pública XlaSparseCoreAdam

Métodos públicos

estático XlaSparseCoreAdam
crear ( alcance alcance , Tabla de incrustación de operando <Float>, índices de operando <Integer>, gradiente de operando <Float>, velocidad de aprendizaje de operando <Float>, impulso de operando <Float>, velocidad de operando <Float>, operando <Float> beta1, operando <Float > beta2, operando <Float> épsilon, ancho de función largo, uso booleanoSumInsideSqrt)
Método de fábrica para crear una clase que envuelve una nueva operación XlaSparseCoreAdam.
Salida <Flotante>
actualizadoEmbeddingTable ()
Salida <Flotante>
actualizadoMomentum ()
Salida <Flotante>
velocidad actualizada ()

Métodos heredados

Métodos públicos

creación estática pública de XlaSparseCoreAdam (alcance del alcance , tabla de incrustación del operando <Float>, índices del operando <Integer>, gradiente del operando <Float>, velocidad de aprendizaje del operando <Float>, impulso del operando <Float>, velocidad del operando <Float>, operando <Float> beta1, Operando <Float> beta2, Operando <Float> épsilon, FeatureWidth largo, uso booleanoSumInsideSqrt)

Método de fábrica para crear una clase que envuelve una nueva operación XlaSparseCoreAdam.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de XlaSparseCoreAdam

Salida pública <Float> actualizadaEmbeddingTable ()

Salida pública <Float> actualizadoMomentum ()

Salida pública <Float> velocidad actualizada ()