numpy.lib.introspect.opt_func_info#

lib.introspect.opt_func_info(func_name=None, signature=None)[source]#

返回一个字典,其中包含所有优化函数的当前支持的 CPU 分派功能。

参数:
func_namestr (可选)

用于按函数名称过滤的正则表达式。

signaturestr (可选)

用于按数据类型过滤的正则表达式。

返回值:
dict

一个字典,其中键是优化的函数名称,值是嵌套字典,表示基于数据类型的支持目标。

示例

检索名为“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)"
        }
      }
    }