mars.tensor.fft.fftshift#
- mars.tensor.fft.fftshift(x, axes=None)[源代码]#
Shift the zero-frequency component to the center of the spectrum.
This function swaps half-spaces for all axes listed (defaults to all). Note that
y[0]
is the Nyquist component only iflen(x)
is even.- 参数
x (array_like) – Input tensor.
axes (int or shape tuple, optional) – Axes over which to shift. Default is None, which shifts all axes.
- 返回
y – The shifted tensor.
- 返回类型
Tensor
参见
ifftshift
The inverse of fftshift.
实际案例
>>> import mars.tensor as mt
>>> freqs = mt.fft.fftfreq(10, 0.1) >>> freqs.execute() array([ 0., 1., 2., 3., 4., -5., -4., -3., -2., -1.]) >>> mt.fft.fftshift(freqs).execute() array([-5., -4., -3., -2., -1., 0., 1., 2., 3., 4.])
Shift the zero-frequency component only along the second axis:
>>> freqs = mt.fft.fftfreq(9, d=1./9).reshape(3, 3) >>> freqs.execute() array([[ 0., 1., 2.], [ 3., 4., -4.], [-3., -2., -1.]]) >>> mt.fft.fftshift(freqs, axes=(1,)).execute() array([[ 2., 0., 1.], [-4., 3., 4.], [-1., -3., -2.]])