Weź pierwszy w Pytorch DataLoader

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