scipy.fft.

set_backend#

scipy.fft.set_backend(backend, coerce=False, only=False)[source]#

設定後端的上下文管理器,作用範圍固定。

當進入 with 語句時,指定的後端將會被加入到可用後端列表中,並具有最高優先權。當離開時,後端會被重置為進入作用範圍之前的狀態。

參數:
backend{object, ‘scipy’}

要使用的後端。可以是包含已知後端名稱 {‘scipy’} 的 str,或是實作 uarray 協議的物件。

coercebool,選用

是否允許對 x 參數進行高成本的轉換。例如,為了 CuPy 後端將 NumPy 陣列複製到 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])