SparseMatrixMul

kelas akhir publik SparseMatrixMul

Perkalian berdasarkan elemen dari matriks renggang dengan tensor padat.

Mengembalikan matriks renggang.

Tensor padat `b` dapat berupa skalar; jika tidak, `a` harus berupa `SparseMatrix` peringkat-3; dalam hal ini `b` harus berbentuk `[ukuran_batch, 1, 1]` dan operasi perkalian disiarkan.

CATATAN meskipun `b` adalah nol, struktur ketersebaran keluaran tidak berubah.

Metode Publik

Keluaran <Objek>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
statis <T> SparseMatrixMul
buat ( Lingkup lingkup, Operan <?> a, Operan <T> b)
Metode pabrik untuk membuat kelas yang membungkus operasi SparseMatrixMul baru.
Keluaran <?>
keluaran ()
Tensor keluaran padat.

Metode Warisan

Metode Publik

Output publik <Objek> asOutput ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

SparseMatrixMul statis publik buat ( Lingkup lingkup, Operan <?> a, Operan <T> b)

Metode pabrik untuk membuat kelas yang membungkus operasi SparseMatrixMul baru.

Parameter
cakupan ruang lingkup saat ini
A Sebuah CSRSparseMatrix.
B Tensor padat.
Kembali
  • contoh baru dari SparseMatrixMul

Keluaran publik <?> keluaran ()

Tensor keluaran padat.