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