fftshift2# ptyrad.utils.math_ops.fftshift2(x)[source]# A wrapper over torch.fft.fftshift for the last 2 dims