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.