Does the method add_edges preserve the batch information? If not, how to do this in an efficient way.
g1 = dgl.graph(([0, 1, 2, 3, 4], [1, 2, 3, 4, 0]))
g2 = dgl.graph(([0, 3, 2, 3, 4], [1, 2, 3, 4, 0]))
bg = dgl.batch([g1, g2])
bg.add_edges([0],[3])
dgl.unbatch(bg)