"examples/git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "da3ffe9bf146c89585ed32aaf3be7f81bea2c4ab"
Commit 02a47c46 authored by rusty1s's avatar rusty1s
Browse files

composite

parent 356d0fe8
......@@ -32,6 +32,8 @@ The package consists of the following operations with reduction types `"sum"|"me
* [**segment_coo**](https://pytorch-scatter.readthedocs.io/en/latest/functions/segment_coo.html) based on sorted indices
* [**segment_csr**](https://pytorch-scatter.readthedocs.io/en/latest/functions/segment_csr.html) based on compressed indices via pointers
In addition, we provide the following **composite functions** which make use of `scatter_*` operations under the hood: :`scatter_std`, `scatter_logsumexp`, `scatter_softmax` and `scatter_log_softmax`.
All included operations are broadcastable, work on varying data types, are implemented both for CPU and GPU with corresponding backward implementations, and are fully traceable.
## Installation
......
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