Pytorch17 complex fft

[vdutell]

[ChairManMeow-SY]

I'm so confusing for line 91: height, width = im_batch.shape[2], im_batch.shape[1]

As the input is in the shape [N, C, H, W], the squeezed im_batch should be in [N, H, W].....

Thus height = im_batch.shape[1] and width = im_batch.shape[2] ?