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.