a_d:

Here, I noticed that It loaded the data of the first and label of the second.


I’m not sure what you mean by that? There is not concept of data and label here. The Dataset should return the right pair of them for a given item.