Isn't this equal to call edge_index(b, a)?