bundles / scipy 1.17.1 / scipy / fft / _basic / fft2
_Function
scipy.fft._basic:fft2
source: /scipy/fft/_basic.py :838
Signature
def fft2 ( x , s = None , axes = (-2, -1) , norm = None , overwrite_x = False , workers = None , * , plan = None ) Summary
Compute the 2-D discrete Fourier Transform
Extended Summary
This function computes the N-D discrete Fourier Transform over any axes in an M-D array by means of the Fast Fourier Transform (FFT). By default, the transform is computed over the last two axes of the input array, i.e., a 2-dimensional FFT.
Parameters
x: array_likeInput array, can be complex
s: sequence of ints, optionalShape (length of each transformed axis) of the output (
s[0]refers to axis 0,s[1]to axis 1, etc.). This corresponds tonforfft(x, n). Along each axis, if the given shape is smaller than that of the input, the input is cropped. If it is larger, the input is padded with zeros. ifsis not given, the shape of the input along the axes specified byaxesis used.axes: sequence of ints, optionalAxes over which to compute the FFT. If not given, the last two axes are used.
norm: {"backward", "ortho", "forward"}, optionalNormalization mode (see fft). Default is "backward".
overwrite_x: bool, optionalIf True, the contents of
xcan be destroyed; the default is False. See fft for more details.workers: int, optionalMaximum number of workers to use for parallel computation. If negative, the value wraps around from
os.cpu_count(). See fft for more details.plan: object, optionalThis argument is reserved for passing in a precomputed plan provided by downstream FFT vendors. It is currently not used in SciPy.
Returns
out: complex ndarrayThe truncated or zero-padded input, transformed along the axes indicated by
axes, or the last two axes ifaxesis not given.
Raises
: ValueErrorIf
sandaxeshave different length, oraxesnot given andlen(s) != 2.: IndexErrorIf an element of
axesis larger than the number of axes ofx.
Notes
fft2 is just fftn with a different default for axes.
The output, analogously to fft, contains the term for zero frequency in the low-order corner of the transformed axes, the positive frequency terms in the first half of these axes, the term for the Nyquist frequency in the middle of the axes and the negative frequency terms in the second half of the axes, in order of decreasingly negative frequency.
See fftn for details and a plotting example, and fft for definitions and conventions used.
Array API Standard Support
fft2 has experimental support for Python Array API Standard compatible backends in addition to NumPy. Please consider testing these features by setting an environment variable SCIPY_ARRAY_API=1 and providing CuPy, PyTorch, JAX, or Dask arrays as array arguments. The following combinations of backend and device (or other capability) are supported.
==================== ==================== ==================== Library CPU GPU ==================== ==================== ==================== NumPy ✅ n/a CuPy n/a ✅ PyTorch ✅ ✅ JAX ✅ ✅ Dask ⚠️ computes graph n/a ==================== ==================== ====================
See
dev-arrayapifor more information.
Examples
import scipy.fft import numpy as np x = np.mgrid[:5, :5][0]✓
scipy.fft.fft2(x)
✗See also
- fft
The 1-D FFT.
- fftn
The N-D FFT.
- fftshift
Shifts zero-frequency terms to the center of the array. For 2-D input, swaps first and third quadrants, and second and fourth quadrants.
- ifft2
The inverse 2-D FFT.
Aliases
-
scipy.fft.fft2