Resize Image Dimensions of a 4D tensor

Hi All,

I have an 4D image tensor of dimension (10, 10, 256, 256) which I want to resize the image height and width to 100 x 100 such that the resulting 4D tensor is of the dimension (10, 10, 100, 100). I have tried several options but not getting the required dimensions. Any suggestions to resolved this are welcomed.

Have you tried AdaptiveAvgPool2d — PyTorch 1.11.0 documentation?