index.rst 711 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
:github_url: https://github.com/rusty1s/pytorch_scatter

PyTorch Scatter documentation
===============================

rusty1s's avatar
rusty1s committed
6
This package consists of a small extension library of highly optimized sparse update (scatter) operations for the use in `PyTorch <http://pytorch.org/>`_, which are missing in the main package.
rusty1s's avatar
typo  
rusty1s committed
7
Scatter operations can be roughly described as reduce operations based on a given "group-index" tensor.
rusty1s's avatar
rusty1s committed
8
9
10

All included operations work on varying data types, are implemented both for CPU and GPU and include a backwards implementation.

rusty1s's avatar
rusty1s committed
11
12
13
.. toctree::
   :glob:
   :maxdepth: 1
rusty1s's avatar
rusty1s committed
14
15
16
   :caption: Package reference

   functions/*
rusty1s's avatar
rusty1s committed
17
18
19
20
21
22

Indices and tables
==================

* :ref:`genindex`
* :ref:`modindex`