scipy.fft.

set_backend#

scipy.fft.set_backend(backend, coerce=False, only=False)[源代码]#

上下文管理器在固定范围内设置后端。

进入 with 语句后，会将给定的后端添加到可用后端列表中，并赋予最高优先级。退出后，后端将重置为进入范围之前的状态。

参数:

backend{object, ‘scipy’}: 要使用后端。可以是包含已知后端 {‘scipy’} 名称的 str，也可以是实现 uarray 协议的对象。
coercebool，可选: 是否允许 expensive 转换x 参数。例如，将 NumPy 数组复制到 CuPy 后端的 GPU。隐含 only。
onlybool，可选: 如果 only 为 True，且此后端返回 NotImplemented，则会立即触发 BackendNotImplemented 错误。忽略任何低优先级的后端。

示例

>>> import scipy.fft as fft
>>> with fft.set_backend('scipy', only=True):
...     fft.fft([1])  # Always calls the scipy implementation
array([1.+0.j])