I’m doing an image processing task and I want to use torch.cat to concat pictures belonging to two different folders. The size of the images in folder 1 is 224 * 224 * 3, and the size of the images in folder 2 is 224 *  224  *  1.Each folder has 100 images.I want them to be concated in the axial direction and I will use the  results as input for deep learning.
How can I write this program?