Pegue o primeiro no Dataloader Pytorch

 inputs, classes = next(iter(dataloader))   
Sleepy Shark