Obtenez un lot unique à partir du chargeur de données Torch

inputs, classes = next(iter(dataloader))
Difficult Dormouse