dgl.unbatch

dgl.unbatch(graph)[source]

Return the list of graphs in this batch.

Parameters:graph (BatchedDGLGraph) – The batched graph.
Returns:A list of DGLGraph objects whose attributes are obtained by partitioning the attributes of the graph. The length of the list is the same as the batch size of graph.
Return type:list

Notes

Unbatching will break each field tensor of the batched graph into smaller partitions.

For simpler tasks such as node/edge state aggregation, try to use readout functions.

See also

batch()