Python PyTorch Datasets Converting entire Dataset to
rliuslxn