tf.compat.v1.sparse_matmul

Умножает матрицу "a" на матрицу "b". (устарела)

tf.compat.v1.sparse_matmul(
    a: Annotated[Any, tf.raw_ops.Any],
    b: Annotated[Any, tf.raw_ops.Any],
    transpose_a: bool = False,
    transpose_b: bool = False,
    a_is_sparse: bool = False,
    b_is_sparse: bool = False,
    name=None
) -> Annotated[Any, tf.raw_ops.Any]

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

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