XlaSparseDenseMatmul

classe finale publique XlaSparseDenseMatmul

Méthodes publiques

Sortie <Flottant>
activations ()
statique XlaSparseDenseMatmul
créer ( Portée de la portée , Operand <Integer> rowIds, Operand <?> colIds, Operand <Float> valeurs, Operand <?> offsets, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaSparseDenseMatmul.
Sortie <Entier>
lignesPointeurs ()
Sortie <Entier>
triésEmbeddingIds ()
Sortie <Flottant>
Gains triés ()
Sortie <Entier>
triésSampleIds ()

Méthodes héritées

Méthodes publiques

Activations <Float> de sortie publique ()

public static XlaSparseDenseMatmul create ( Scope scope, Operand <Integer> rowIds, Operand <?> colIds, Operand <Float> valeurs, Operand <?> offsets, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération XlaSparseDenseMatmul.

Paramètres
portée portée actuelle
Retour
  • une nouvelle instance de XlaSparseDenseMatmul

sortie publique <Integer> rowPointers ()

sortie publique <Integer> sortedEmbeddingIds ()

sortie publique <Float> triésGains ()

sortie publique <Integer> sortedSampleIds ()