Unverified Commit c0e1a8b6 authored by mversionpenny's avatar mversionpenny Committed by GitHub
Browse files

Changes in the documentation of segment_csr

Hi,
I think the top index of the sum is not right in the documentation of segment_csr. I might be wrong though!
parent 335525a7
...@@ -64,7 +64,7 @@ def segment_csr(src: torch.Tensor, indptr: torch.Tensor, ...@@ -64,7 +64,7 @@ def segment_csr(src: torch.Tensor, indptr: torch.Tensor,
.. math:: .. math::
\mathrm{out}_i = \mathrm{out}_i =
\sum_{j = \mathrm{indptr}[i]}^{\mathrm{indptr}[i+i]}~\mathrm{src}_j. \sum_{j = \mathrm{indptr}[i]}^{\mathrm{indptr}[i+1]-1}~\mathrm{src}_j.
Due to the use of index pointers, :meth:`segment_csr` is the fastest Due to the use of index pointers, :meth:`segment_csr` is the fastest
method to apply for grouped reductions. method to apply for grouped reductions.
......
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