pytorch dataloader get one item code example Example: take the first in dataloader pytorch inputs, classes = next(iter(dataloader))