XlaSparseDenseMatmul

clase final pública XlaSparseDenseMatmul

Métodos públicos

Salida <Flotante>
activaciones ()
estático XlaSparseDenseMatmul
crear (alcance del alcance , Operando <Integer> rowIds, Operando <?> colIds, Operando <Float> valores, Operando <?> compensaciones, Operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Método de fábrica para crear una clase que envuelve una nueva operación XlaSparseDenseMatmul.
Salida <Entero>
punteros de fila ()
Salida <Entero>
ordenadosEmbeddingIds ()
Salida <Flotante>
ganancias ordenadas ()
Salida <Entero>
ordenadosSampleIds ()

Métodos heredados

Métodos públicos

Activaciones de salida pública <Float> ()

creación pública estática de XlaSparseDenseMatmul (alcance del alcance , operando <Integer> rowIds, operando <?> colIds, valores de operando <Float>, compensaciones de operando <?>, operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

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

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

Salida pública <Entero> punteros de fila ()

Salida pública <Entero> sortedEmbeddingIds ()

Salida pública <Float> sortedGains ()

Salida pública <Entero> sortedSampleIds ()