mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
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
(cherry picked from commit 225929ecf2)
This commit is contained in:
parent
896703d3d7
commit
8cdcc1181c
|
|
@ -590,6 +590,7 @@ Torch functions specific to sparse Tensors
|
|||
sparse_csr_tensor
|
||||
sparse.sum
|
||||
sparse.addmm
|
||||
sparse.sampled_addmm
|
||||
sparse.mm
|
||||
sspaddmm
|
||||
hspmm
|
||||
|
|
|
|||
|
|
@ -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`
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user