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