XlaSparseDenseMatmul

kelas akhir publik XlaSparseDenseMatmul

Metode Publik

Keluaran <Mengambang>
XlaSparseDenseMatmul statis
buat ( Lingkup cakupan , Operand <Integer> rowIds, Operand <?> colIds, Nilai Operand <Float>, Operand <?> offset, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSparseDenseMatmul baru.
Keluaran <Bilangan Bulat>
Keluaran <Bilangan Bulat>
Keluaran <Mengambang>
Keluaran <Bilangan Bulat>

Metode Warisan

Metode Publik

aktivasi Output <Float> publik ()

pembuatan XlaSparseDenseMatmul statis publik ( Lingkup cakupan, Operand <Integer> rowIds, Operand <?> colIds, nilai Operand <Float>, Operand <?> offset, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaSparseDenseMatmul baru.

Parameter
cakupan ruang lingkup saat ini
Kembali
  • contoh baru dari XlaSparseDenseMatmul

Output publik <Bilangan Bulat> rowPointer ()

Output publik <Bilangan Bulat> diurutkanEmbeddingIds ()

Output publik <Float> diurutkanGains ()

Output publik <Bilangan Bulat> diurutkanSampleIds ()