Add missing entry for sampled_addmm in sparse.rst (#72312)

Summary:
Let's make the documentation for `torch.sparse.sampled_addmm` searchable in the PyTorch documentation.
This PR shall be cherry-picked for the next 1.11 release.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/72312

Reviewed By: davidberard98

Differential Revision: D34045230

Pulled By: cpuhrsch

fbshipit-source-id: c1b1dc907443284857f48c8ce1efab22c6701bbe
This commit is contained in:
Ivan Yashchuk 2022-02-07 15:57:37 -08:00 committed by Facebook GitHub Bot
parent e464aa91eb
commit 225929ecf2
2 changed files with 3 additions and 0 deletions

View File

@ -590,6 +590,7 @@ Torch functions specific to sparse Tensors
sparse_csr_tensor
sparse.sum
sparse.addmm
sparse.sampled_addmm
sparse.mm
sspaddmm
hspmm

View File

@ -98,7 +98,9 @@ Performs a matrix multiplication of the dense matrices :attr:`mat1` and :attr:`m
specified by the sparsity pattern of :attr:`input`. The matrix :attr:`input` is added to the final result.
Mathematically this performs the following operation:
.. math::
\text{out} = \alpha\ (\text{mat1} \mathbin{@} \text{mat2})*\text{spy}(\text{input}) + \beta\ \text{input}
where :math:`\text{spy}(\text{input})` is the sparsity pattern matrix of :attr:`input`, :attr:`alpha`