The method of masking edge

[StefanIsSmart]

For example, Some graph is directed (like [1,0],[0,1]), and I want to mask the double directed edges together randomly.
How could I do that? Is there any function?

[rusty1s]

mask = (edge_index[0] == node_id) & (edge_index[1] == node_id)
edge_index = edge_index[:, ~mask]