← all repositories

rusty1s/pytorch_scatter

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

1.7k stars Python ML Frameworks
pytorch_scatter
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.