Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/nerfacc.git" did not exist on "e6647a003f37a7141d0d8b8ec63770ee1c59b724"
Commit 3b2dee22 authored by rusty1s's avatar rusty1s
Browse files

added description

parent c8893e9e
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 2 deletions
README.md
View file @ 3b2dee22
...@@ -11,6 +11,9 @@ ...@@ -11,6 +11,9 @@
-------------------------------------------------------------------------------- --------------------------------------------------------------------------------
This package consists of a small extension library of highly optimised sparse update operations for the use in [PyTorch](http://pytorch.org/).
It consists of the following operations:
* `scatter_add` * `scatter_add`
* `scatter_sub` * `scatter_sub`
* `scatter_mul` * `scatter_mul`
...@@ -19,13 +22,15 @@ ...@@ -19,13 +22,15 @@
* `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.
## Installation ## Installation
```sh ```sh
python setup.py install python setup.py install
``` ```
## Usage ## Example
```py ```py
from torch_scatter import scatter_max from torch_scatter import scatter_max
......
setup.cfg
View file @ 3b2dee22
[metadata]
description-file = README.md
[aliases] [aliases]
test=pytest test = pytest
[tool:pytest] [tool:pytest]
addopts = --capture=no --cov addopts = --capture=no --cov
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