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" должны быть Tensors, а не SparseTensors. Этот оператор оптимизирован для случая, когда по крайней мере одна из матриц "a" или "b" является разреженной, в том смысле, что у них большая доля нулевых значений. Критерий использования этого оператора вместо умножения плотных матриц на одной платформе составлял 30% нулевых значений в разреженной матрице.

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