Note
🚧 EXPERIMENTAL and IN DEVELOPMENT: It's no where near from a complete GNN library (like PyTorch Geometric, TensorFlow GNN and others. It's being populated and the need arises.
Contributions are most welcome. 🚧
geometry.RadiusEdges: returns edges between the source and target points that are within a give radius. It works for arbitrary dimensions (2D, 3D, etc.).geometry.NearestEdges: returns the edges between each source point and its closest target point. It works for arbitrary dimensions (2D, 3D, etc.).graph.UnionEdges: returns the union from a list of edge sets.graph.SortEdgesBySource: sort edges by source id.layers.SparseSoftmax: calculating a Softmax on a sparse vector (typically index by some set of edge indices).