pytorch, 遍历dataloade时，如何获取原始index

pytorch，遍历dataloader的时候

train_loader = DataLoader(train_set, batch_size=batch_size_train, shuffle=True)

因为dataloader已经被shuffle过，

如何知道，他在原始dataset中的

index

torch.utils.data — PyTorch 1.9.1 documentation