Map indices from line graph back to the original graph

[lisiq]

Hi everyone!
I am using LineGraph, and I was wondering if there is a way to map back from node indices of the line graph to node indices of the nodes in the original graph (ie. index of node v in the line graph maps to the node indices of u and w respectively)?
Thank you in advance!

[EdisonLeeeee]

The nodes in the line graph were mapped to the edge_index of the original graphs, so you can simply run:

# v: index of node in the line graph
# u, w: index of node in the original graph
# edge_index: edge indices of original graph

u, w = edge_index[v]

[lisiq]

Thank you!