轻松鲜花

文章
6
资源
0
加入时间
3年0月21天

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

pytorch,遍历dataloader的时候train_loader = DataLoader(train_set, batch_size=batch_size_train, shuffle=True)因为dataloader已经被shuffle过,如何知道,他在原始dataset中的indextorch.utils.data — PyTorch 1.9.1 documentation...