How can I merge several graphs into a giant graph?

[chenyuqi990215]

I have a graph with edge_index shape [2, e], and I would like to construct a batched graph with the same edge_index to get a giant graph with edge_index shape [2, b*e], where b is the batch_size.

The function may look like the following:

def merge_graphs(edge_index, b)

Is there any simple solution to the function?

[rusty1s]

It's probably the easiest way to wrap them int Data objects:

data_list = [Data(edge_index=edge_index), ...]
edge_index = Batch.from_data_list(data_list).edge_index