Created by: BozianuLeon
Simple alternative to knn imagining a sphere around each triggered DOM, taken identically from torch_geometric.nn.