scipy.fft.set_workers

scipy.fft.set_workers(workers)

Context manager for the default number of workers used in scipy.fft

Parameters
workersint

The default number of workers to use

Examples

>>> from scipy import fft, signal
>>> rng = np.random.default_rng()
>>> x = rng.standard_normal((128, 64))
>>> with fft.set_workers(4):
...     y = signal.fftconvolve(x, x)