rusty1s/pytorch_scatter
A PyTorch extension library providing highly optimized sparse scatter and segment operations with reduction types sum, mean, min, and max.

Velocity · 7d
+0.6
★ / day
Trend
→steady
star history
This package provides optimized sparse update operations (scatter and segment) for PyTorch that are missing in the main library. Scatter operations perform reduce operations based on group-index tensors without requiring sorted indices, while segment operations work on sorted index tensors. The library is used extensively in graph neural networks and other deep learning applications requiring sparse tensor aggregations.