image1140×412 111 KB

In this image, we  concat three images on the batch dimension to get the output as shown in the image. I hope this visual representation helps.
First we had three images with each [1, c,h,w]. After concat we got [3,c,h,w]