torch.utils.data.DataLoader "next" function?

2

try this

batch = next(iter(dataloader))

input, target = batch

in this way i believe you can only access a single batch of data, this is more
effecient than for loop if you want to view a single batch of data