Unverified Commit 1d1b08b0 authored by czkkkkkk's avatar czkkkkkk Committed by GitHub
Browse files

[Sparse] Fix sddmm docstring (#5169)

parent 85bd0cde
...@@ -11,9 +11,9 @@ def sddmm( ...@@ -11,9 +11,9 @@ def sddmm(
) -> SparseMatrix: ) -> SparseMatrix:
r"""Sampled-Dense-Dense Matrix Multiplication (SDDMM). r"""Sampled-Dense-Dense Matrix Multiplication (SDDMM).
``sddmm`` multiplies two dense matrices :attr:`mat1` and :attr:`mat2` ``sddmm`` matrix-multiplies two dense matrices :attr:`mat1` and :attr:`mat2`
at the nonzero locations of sparse matrix :attr:`A`. Values of :attr:`A` , then elementwise-multiplies the result with sparse matrix :attr:`A` at the
is not considered during the computation. nonzero locations.
Mathematically ``sddmm`` is formulated as: Mathematically ``sddmm`` is formulated as:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment