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

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`