Fft (shifted) in pytorch for lena image

I am looking for a minimum example to plot the magnitude and phase of the popular lena image in pytorch. Essentially, I want to replicate this (the mag and phase plots) from matlab:

is there a minimum example somewhere I can borrow for this?