Get index for k-hop neighbors separately using neighbor loader

[RX28666]

Hello,

I am using NeighborLoader to do sampling, and using data.n_id = torch.arange(data.num_nodes) to get index for targeted nodes and their neighbors.
I am wondering if there is a way to get the index for one hop, second hop, etc. respectively. So I am able to know which nodes are the first/second hop neighbor of the targeted nodes.
Thanks.

[LukeLIN-web]

I also have this problem.

[rusty1s]

We are working on integrating this here: pyg-team/pyg-lib#197