Pytorch Dataloader Get Data Back