rusty1s / pytorch_scatter Python 1.8K 206 PyTorch Extension Library of Optimized Scatter Operations pytorch scatter segment gather