kelas akhir publik XlaSparseDenseMatmul
Metode Publik
Keluaran <Mengambang> | aktivasi () |
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> | barisPointer () |
Keluaran <Bilangan Bulat> | |
Keluaran <Mengambang> | |
Keluaran <Bilangan Bulat> |
Metode Warisan
Metode 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