Obtenha um único lote do carregador de dados da tocha
inputs, classes = next(iter(dataloader))
Difficult Dormouse
inputs, classes = next(iter(dataloader))