I am looking for the index of each image, I mean the index here:
def __getitem__(self, index):    
            x1, y1 = self.Train[index]
            x2, y2 = self.Test[index]
            
            x = torch.stack((x1,x2))
            y = torch.stack((torch.tensor(y1), torch.tensor(y2)))
            
            return x, y, index

but this index is for concatenated images, if I am correct? I want to track the index of each image apart from their concatenation.