XlaSparseDenseMatmul

publiczna klasa końcowa XlaSparseDenseMatmul

Metody publiczne

Wyjście <Pływak>
aktywacje ()
statyczny XlaSparseDenseMatmul
utwórz ( Zakres zakresu, Operand <Integer> rowIds, Operand <?> colIds, Argument <Float> wartości, Operand <?> przesunięcia, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Długi inputSize)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaSparseDenseMatmul.
Wyjście <liczba całkowita>
rowPointers ()
Wyjście <liczba całkowita>
sortowaneEmbeddingIds ()
Wyjście <Pływak>
posortowaneZyski ()
Wyjście <liczba całkowita>
sortowaneIdSampleIds ()

Metody dziedziczone

Metody publiczne

publiczne aktywacje wyjścia <Float> ()

public static XlaSparseDenseMatmul create ( Zakres zakresu , Operand <Integer> rowIds, Operand <?> colIds, Argument <Float> wartości, Operand <?> przesunięcia, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaSparseDenseMatmul.

Parametry
zakres aktualny zakres
Zwroty
  • nowa instancja XlaSparseDenseMatmul

publiczne dane wyjściowe <Integer> rowPointers ()

publiczne wyjście <Integer> sortedEmbeddingIds ()

publiczne wyjście <Float> sortedGains ()

public Output <Integer> sortedSampleIds ()