Efficient fft2 in pytorch

Hi, I have a tensor x whose shape is 233, and I want to do fft2 to x with size 4*4. This means that do right and down zeros padding to x[0,:,:], x[1,:,:], then do fft2 to them respectively. I wonder how can I realize above efficiently. Thank you for your apply.