numpy.lib.introspect.opt_func_info#
- lib.introspect.opt_func_info(func_name=None, signature=None)[source]#
返回一個字典,其中包含所有最佳化函式目前支援的 CPU 調度功能。
- 參數:
- func_name字串 (選填)
用於依函式名稱篩選的正規表示式。
- signature字串 (選填)
用於依資料類型篩選的正規表示式。
- 返回:
- 字典
一個字典,其中鍵是最佳化函式名稱,值是巢狀字典,指示基於資料類型支援的目標。
範例
檢索名為 ‘add’ 或 ‘sub’ 的函式以及 ‘float64’ 或 ‘float32’ 資料類型的調度資訊
>>> import numpy as np >>> dict = np.lib.introspect.opt_func_info( ... func_name="add|abs", signature="float64|complex64" ... ) >>> import json >>> print(json.dumps(dict, indent=2)) { "absolute": { "dd": { "current": "SSE41", "available": "SSE41 baseline(SSE SSE2 SSE3)" }, "Ff": { "current": "FMA3__AVX2", "available": "AVX512F FMA3__AVX2 baseline(SSE SSE2 SSE3)" }, "Dd": { "current": "FMA3__AVX2", "available": "AVX512F FMA3__AVX2 baseline(SSE SSE2 SSE3)" } }, "add": { "ddd": { "current": "FMA3__AVX2", "available": "FMA3__AVX2 baseline(SSE SSE2 SSE3)" }, "FFF": { "current": "FMA3__AVX2", "available": "FMA3__AVX2 baseline(SSE SSE2 SSE3)" } } }