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