scipy.fft.

set_backend

scipy.fft.set_backend(backend, coerce=False, only=False)

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

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

参数:

backend{object, ‘scipy’}

要使用的后端。可以是包含已知后端名称 {'scipy'} 的 str，也可以是实现 uarray 协议的对象。

coercebool，可选

是否允许对 x 参数进行昂贵的转换。例如，将 NumPy 数组复制到 GPU 以用于 CuPy 后端。意味着 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])

返回在新标签页中打开