scipy.fft.
set_backend#
- scipy.fft.set_backend(backend, coerce=False, only=False)[source]#
Context manager to set the backend within a fixed scope.
Upon entering the
withstatement, the given backend will be added to the list of available backends with the highest priority. Upon exit, the backend is reset to the state before entering the scope.- Parameters:
- backend{object, ‘scipy’}
The backend to use. Can either be a
strcontaining the name of a known backend {‘scipy’} or an object that implements the uarray protocol.- coercebool, optional
Whether to allow expensive conversions for the
xparameter. e.g., copying a NumPy array to the GPU for a CuPy backend. Impliesonly.- onlybool, optional
If only is
Trueand this backend returnsNotImplemented, then a BackendNotImplemented error will be raised immediately. Ignoring any lower priority backends.
Examples
>>> import scipy.fft as fft >>> with fft.set_backend('scipy', only=True): ... fft.fft([1]) # Always calls the scipy implementation array([1.+0.j])