separate_graphs

separate_graphs(graph_batch)

Split a batch of collated Graph into a list of single Graphs.

It reverses the operation of collate_graphs().

Parameters:

graph_batch (Graph) – A Graph whose current_shape and true_shape are batched.

Returns:

List of Graphs, each corresponding to one element in the batch.

Return type:

list[Graph]