Commit c692eb09 authored by rusty1s's avatar rusty1s
Browse files

fixed version call, added readme

parent d258f62a
...@@ -11,8 +11,9 @@ ...@@ -11,8 +11,9 @@
-------------------------------------------------------------------------------- --------------------------------------------------------------------------------
This package consists of a small extension library of highly optimised sparse update operations for the use in [PyTorch](http://pytorch.org/). 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.
It consists of the following operations: Scatter-operations can be roughly described as reduce-operations based on a given "group-index" tensor.
The package consists of the following operations:
* `scatter_add` * `scatter_add`
* `scatter_sub` * `scatter_sub`
...@@ -22,7 +23,7 @@ It consists of the following operations: ...@@ -22,7 +23,7 @@ It consists of the following operations:
* `scatter_min` * `scatter_min`
* `scatter_max` * `scatter_max`
All included `scatter` ops work on varying data types, are implemented both for CPU and GPU and include a backwards implementation. All included operations work on varying data types, are implemented both for CPU and GPU and include a backwards implementation.
## Installation ## Installation
......
...@@ -3,6 +3,11 @@ ...@@ -3,6 +3,11 @@
PyTorch Scatter documentation PyTorch Scatter documentation
=============================== ===============================
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.
.. toctree:: .. toctree::
:glob: :glob:
:maxdepth: 1 :maxdepth: 1
......
from os import path as osp from os import path as osp
from setuptools import setup, find_packages from setuptools import setup, find_packages
filename = osp.join(osp.dirname(osp.abspath('__file__')), 'VERSION') filename = osp.join(osp.dirname(__file__), 'VERSION')
with open(filename, 'r') as f: with open(filename, 'r') as f:
version = f.read().strip() version = f.read().strip()
......
...@@ -8,7 +8,7 @@ from .functions.mean import scatter_mean_, scatter_mean ...@@ -8,7 +8,7 @@ from .functions.mean import scatter_mean_, scatter_mean
from .functions.max import scatter_max_, scatter_max from .functions.max import scatter_max_, scatter_max
from .functions.min import scatter_min_, scatter_min from .functions.min import scatter_min_, scatter_min
filename = osp.join(osp.dirname(osp.abspath('__file__')), 'VERSION') filename = osp.join(osp.dirname(__file__), '..', 'VERSION')
with open(filename, 'r') as f: with open(filename, 'r') as f:
__version__ = f.read().strip() __version__ = f.read().strip()
......
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