Numpy FFTN equivalent in Pytorch

I want to perform up to a four-dimensional Fourier transform on a Pytorch Tensor.

What is the equivalent of the function of numpy.fft.fftn(a, axes=(1, 2)) in Pytorch?