How to quickly find indices of edges that contain certain nodes from edge_index?

[scottshufe]

Ask for help. My problem is just as the title described.
For example:
Assume edge_index=[[1, 2, 2, 3, 3], [2, 4, 5, 2, 5]], and I want to find the indices of edges that contain node_list=[2, 3].
What's the quick solution for this question (without any for loops) ?

[EdisonLeeeee]

You can take a look at torch_geometric.utils.subgraph.

[scottshufe]

Thank you so much! It solves my problem perfectly.

[rusty1s]

Please also see the discussion here: #5595

[scottshufe]

Thanks. But here #5593 is here 😂

[rusty1s]

Sorry, I meant #5595.

[scottshufe]

Thank you so much.