tensorflow::ops::SparseMatMul

#include <math_ops.h>

Multiply матрицы "a" и "b".

Краткое описание

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

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

Аргументы:

• scope: Объект Scope

Возвращает:

• 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
)