[Python] Can PyTorch `scatter` or `gather` be used to reproduce `torch_geometric` aggregation...

I can't understand if torch.scatter or torch.gather could be used to reduce values of a tensor according to a reduction function over specified indices.

I've frequently used the torch_geometric.nn.aggr` module's functions in order to perform aggregations over indices, and I want to reproduce these very useful functions using pure PyTorch operations.

Continue reading...