I did not get your point. Since the label of the 4 images respectively is tensor([1, 3, 0, 1]). So, how indexing through label is changing the output? lable[1]-> 1, lable[2]->3, lable[3]->0, label[4]-> 1. Should not be this output. Can you please explain?