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.