Solved by albanD in post #2 
              
              
                Hi, 
Yes, MaxPool2d is built to work with images which are 3D Tensors for a single image and 4D tensors for a batch of images.