Why nn.MaxPool2d cannot work on rank 2 tensor?

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.

1 Like