Python numpy ufunc.signature用法及代码示例

本文简要介绍 python 语言中 numpy.ufunc.signature 的用法。

用法:

ufunc.signature

广义 ufunc 操作的核心元素的定义。

签名确定每个输入/输出数组的维度如何拆分为核心维度和循环维度：

1. 签名中的每个维度都与对应的passed-in 数组的维度匹配，从形状元组的末尾开始。

2. 分配给签名中相同标签的核心尺寸必须具有完全匹配的尺寸，不执行广播。

3. 从所有输入中删除核心维度，其余维度一起广播，定义循环维度。

注意：

广义的 ufunc 在内部用于许多 linalg 函数和测试套件中；下面的例子取自这些。对于在标量上操作的 ufunc，签名为 None，相当于每个参数的“()”。

例子：

>>> np.core.umath_tests.matrix_multiply.signature
'(m,n),(n,p)->(m,p)'
>>> np.linalg._umath_linalg.det.signature
'(m,m)->()'
>>> np.add.signature is None
True  # equivalent to '(),()->()'