tf.raw_ops.SparseMatMul

Умножение матрицы "a" на матрицу "b".

tf.raw_ops.SparseMatMul(
    a, b, transpose_a=False, transpose_b=False, a_is_sparse=False,
    b_is_sparse=False, name=None
)

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

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