scipy.fft.set_backend¶

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

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

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

参数

backend{object，‘scipy’}: 要使用的后端。可以是一个 str 包含已知后端{‘scipy’}或实现uarray协议的对象的名称。
coerce布尔值，可选: 是否允许昂贵的 x 参数。例如将NumPy数组复制到用于CuPy后端的GPU。暗示 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])

scipy.fft.get_workers

scipy.fft.skip_backend