Can I get the index of the sampled node in the original data after using NeighborLoader for sampling

[Danieltobesixabs]

Hello，I've been reproducing a piece of paper recently，when i use the NeighborLoader to sample, I need to get the index of the sampled node in the original data after using NeighborLoader for sampling, but i read the documentation, i found that i can`t get it. So could you tell me how can i get the index？

[rusty1s]

You can add the information to the original data:

data.n_id = torch.arange(data.num_nodes)

[Danieltobesixabs]

Thank you for help, i solved this problem!