I would rather write a custom Dataset and return these three images in your __getitem__ method:
def __getitem__(self, index):
    # load the images and your labels according to your code logic
    ...
    reutrn image1, image2, image3, target
    # or concatenate the images and return them as a single one
    return images, target




 styx97:

does the model understand it’s actually three images and treat it as such ?


I’m not sure to understand this question properly.
Your current code snippet will work, but I’m not sure what your use case is.