python - numpy.ufunc.identity 的使用

Question

我想知道 r = ufct.identity 在函数 ufunc_reduce 中做了什么。它是否只是像 r = 0 一样初始化操作？

a = np.array([2,3,4,5])
b = np.array([8,5,4])
c = np.array([5,4,6,8,3])

def ufunc_reduce(ufct, *vectors):
    vs = np.ix_(*vectors)
    r = ufct.identity
    for v in vs:
        r = ufct(r,v)
    return r

ufunc_reduce(np.add,a,b,c)

Answer 1

是的，对于 ufunc np.add，ufct.identity 是 0。但对于另一个功能，它可能是别的东西。例如，np.multiply.identity 是 1。

您显示的 ufunc_reduce 函数事先并不知道它将被赋予什么功能，因此不能硬编码为使用 0 或 1。它通过检查 ufunc 的 identity 属性来获取正确的值。