тензорный поток:: опс:: SparseMatMul

Умножьте матрицу «а» на матрицу «б».

Краткое содержание

Входные данные должны быть двумерными матрицами, а внутреннее измерение «a» должно соответствовать внешнему измерению «b». И «a», и «b» должны быть Tensor , а не SparseTensor . Эта операция оптимизирована для случая, когда хотя бы одно из «a» или «b» является разреженным, в том смысле, что они имеют большую долю нулевых значений. Безубыточность использования этого метода по сравнению с умножением плотной матрицы на одной платформе составила 30% нулевых значений в разреженной матрице.

Вычисление градиента этой операции будет использовать преимущества разреженности входного градиента только в том случае, если этот градиент исходит от Relu .

Аргументы:

• область: объект области.

Возврат:

• Output : Тензор произведения.

Публичные атрибуты

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
)