scipy.fft.

set_backend

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

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

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

参数:

后端{object, ‘scipy’}

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

强制bool, 可选

是否允许 x 参数进行昂贵的转换。例如，将 NumPy 数组复制到 GPU 以用于 CuPy 后端。意味着 only。

仅bool, 可选

如果 True 并且此后端返回 NotImplemented ，则会立即引发 BackendNotImplemented 错误。忽略任何较低优先级的后端。

示例

Try it in your browser!

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

Go BackOpen In Tab