If you concatenate the images, you’ll get “less” samples, so I’m not sure how you would like to keep the batch size as 6.
Could you explain your use case a bit?
Since you are now dealing with multi-hot encoded targets (i.e. multi-label classification), you could use nn.BCELoss or nn.BCEWithLogitsLoss instead.