def fftshift(a, axes=None): '''Swap half-spaces End up with zero frequency component at centre position axes -- list of axes to swap and if None then all axes swapped ''' return _fft.fftshift(a, axes)
def ifftshift(a, axes=None): '''Swaps back half-spaces End up with zero frequency component is at origin axes -- list of axes to swap and if None then all axes swapped ''' return _fft.ifftshift(a, axes)
def fftn(a, s=None, axes=None): '''Perform nD FFT of shape s along given axes''' return _fft.fftn(a, s, axes)
def ifftn(a, s=None, axes=None): '''Perform inverse nD FFT of shape s along given axes''' return _fft.ifftn(a, s, axes)
def ifft2(a, s=None, axes=(-2,-1)): '''Perform inverse 2D FFT of shape s along given axes''' return _fft.ifft2(a, s, axes)
def fft2(a, s=None, axes=(-2,-1)): '''Perform 2D FFT of shape s along given axes''' return _fft.fft2(a, s, axes)
def ifft(a, n=None, axis=-1): '''Perform inverse 1D FFT along given axis''' axis = a.checkAxis(axis) if n == None: n = a.shape[axis] return _fft.ifft(a, n, axis)
def fftfreq(n, d=1.0): '''Sample frequencies for DFT n -- number of samples d -- sample spacing ''' return _fft.sampleFrequencies(n, d)
def fft(a, n=None, axis=-1): '''Perform 1D FFT along given axis''' axis = a.checkAxis(axis) if n is None: n = a.shape[axis] return _fft.fft(a, n, axis)