XlaSparseDenseMatmul

genel final dersi XlaSparseDenseMatmul

Genel Yöntemler

Çıkış <Kamanma>
aktivasyonlar ()
statik XlaSparseDenseMatmul
create ( Kapsam kapsamı, İşlenen <Tamsayı> satır kimlikleri, İşlenen <?> colId'ler, İşlenen <Float> değerleri, İşlenen <?> ofsetleri, İşlenen <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Yeni bir XlaSparseDenseMatmul işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Çıkış <Tamsayı>
rowPointers ()
Çıkış <Tamsayı>
sortedEmbedddingIds ()
Çıkış <Kamanma>
sıralanmışKazançlar ()
Çıkış <Tamsayı>
sıralanmışÖrnek Kimlikleri ()

Kalıtsal Yöntemler

Genel Yöntemler

genel Çıkış <Float> aktivasyonları ()

public static XlaSparseDenseMatmul create ( Kapsam kapsamı, İşlenen <Tamsayı> satır kimlikleri, İşlenen <?> colId'ler, İşlenen <Float> değerleri, İşlenen <?> ofsetleri, İşlenen <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Yeni bir XlaSparseDenseMatmul işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.

Parametreler
kapsam mevcut kapsam
İadeler
  • XlaSparseDenseMatmul'un yeni bir örneği

public Çıkış <Tamsayı> rowPointers ()

public Output <Tamsayı> sortedEmbeddingIds ()

public Output <Float> sortedGains ()

genel Çıkış <Tamsayı> sortedSampleIds ()