🚀 The feature, motivation and pitch

I propose to support to_batch_edge_index similar to unbatch_edge_index. The input could be a list of edge_index tensors that will be merged into one edge_index and the output an edge_index and a batch vector.

Alternatives

No response

Additional context

No response