search

schrojunzhang / KarmaDock

https://www.nature.com/articles/s43588-023-00511-5
Apache License 2.0
92 stars 12 forks source link

Why not use torch.scatter_reduce_ instead of the third-party pytorch_scatter operator #14

Open GuWei007 opened 9 months ago

GuWei007 commented 9 months ago

https://github.com/rusty1s/pytorch_scatter

https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_reduce_.html#torch.Tensor.scatter_reduce_

zhongxiang117 commented 9 months ago

Third party modules torch_* (where * means scatter, cluster, sparse, spline, etc.) are well maintained and documented, also are widely used as a complement of the PyTorch module. They are more like specific utilities which meet the zen of Python that one module only focusing on one thing and doing it well. You will learn their benefits when you go deeper programming.