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)"
        }
      }
    }