tensorflow:: ops:: SparseMatMul

Pomnóż macierz „a” przez macierz „b”.

Streszczenie

Dane wejściowe muszą być macierzami dwuwymiarowymi, a wymiar wewnętrzny „a” musi odpowiadać wymiarowi zewnętrznemu „b”. Zarówno „a”, jak i „b” muszą być Tensor , a nie typu SparseTensor . Ta operacja jest zoptymalizowana pod kątem sytuacji, gdy co najmniej jedna z macierzy „a” lub „b” jest rzadka, w tym sensie, że mają dużą proporcję wartości zerowych. Próg rentowności dla użycia tego w porównaniu z gęstą macierzą mnożenia na jednej platformie wynosił 30% wartości zerowych w macierzy rzadkiej.

Obliczenia gradientu w tej operacji wykorzystają jedynie rzadkość gradientu wejściowego, jeśli gradient ten pochodzi z Relu .

Argumenty:

• zakres: obiekt zakresu

Zwroty:

• Output : Tensor iloczynu.

Atrybuty publiczne

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
)