tf.compat.v1.sparse_matmul

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

tf.compat.v1.sparse_matmul(
    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.