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])