scipy.fft.

set_backend#

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

在固定作用域内设置后端的上下文管理器。

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

参数:
backend{object, ‘scipy’}

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

coerce布尔值, 可选

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

only布尔值, 可选

如果 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])
本页内容