aliran tensor:: operasi:: JarangMatMul

Kalikan matriks "a" dengan matriks "b".

Ringkasan

Inputnya harus berupa matriks dua dimensi dan dimensi dalam "a" harus sesuai dengan dimensi luar "b". Baik "a" maupun "b" harus berupa Tensor , bukan SparseTensor . Operasi ini dioptimalkan untuk kasus di mana setidaknya salah satu dari "a" atau "b" jarang, dalam arti bahwa keduanya memiliki sebagian besar nilai nol. Titik impas untuk menggunakan ini versus perkalian matriks padat pada satu platform adalah 30% nilai nol dalam matriks renggang.

Komputasi gradien pada operasi ini hanya akan memanfaatkan ketersebaran pada gradien masukan jika gradien tersebut berasal dari Relu .

Argumen:

• ruang lingkup: Objek Lingkup

Pengembalian:

• Output : Tensor produk.

Atribut publik

Operation operation

::tensorflow::Output product

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const SparseMatMul::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AIsSparse(
  bool x
)

Attrs BIsSparse(
  bool x
)

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)