We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 82f13f5 commit 121263fCopy full SHA for 121263f
torch/sparse/__init__.py
@@ -29,8 +29,8 @@ def mm(mat1, mat2):
29
r"""
30
Performs a matrix multiplication of the sparse matrix :attr:`mat1`
31
and dense matrix :attr:`mat2`. Similar to :func:`torch.mm`, If :attr:`mat1` is a
32
- (n \times m)(n×m) tensor, :attr:`mat2` is a (m \times p)(m×p) tensor, out will be a
33
- (n \times p)(n×p) dense tensor. :attr:`mat1` need to have `sparse_dim = 2`.
+ :math:`(n \times m)` tensor, :attr:`mat2` is a :math:`(m \times p)` tensor, out will be a
+ :math:`(n \times p)` dense tensor. :attr:`mat1` need to have `sparse_dim = 2`.
34
This function also supports backward for both matrices. Note that the gradients of
35
:attr:`mat1` is a coalesced sparse tensor.
36
0 commit comments