Batch Objects

Batch Objects#

GraphDataBatch(graphs, **kwargs)

Concatenates multiple GraphData into a single unified GraphDataBatch for efficient computation and parallelization over multiple graphs.

batch(graphs)

Constructs a GraphDataBatch object from a list of GraphData.

unbatch(batch)

Reconstructs a list of GraphData objects from a GraphDataBatch.

collate(graph_list)

Concatenates attributes of multiple graphs based on the specifications of each GraphData.