numpy.lib.introspect.opt_func_info

lib.introspect.opt_func_info(func_name=None, signature=None)

返回一个包含当前所有优化函数支持的 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)"
        }
      }
    }