GraphData

Graph in COO format for GNN layers.

Record fields

NameDescription
BatchOptional batch vector [N] mapping each node to its graph index.
EdgeAttrOptional edge features [E, D].
EdgeIndexEdge index in COO format [2, E]. Row 0 = source, row 1 = target.
XNode feature matrix [N, F].