NumPy中+运算符如何触发__add__方法？从+实现理解广播机制

一、+触发__add__的底层逻辑 #

这是Python的运算符重载规则决定的：当你写a + b时，Python解释器会自动调用a.__add__(b)方法——只要a所属的类实现了__add__，+就会被映射到这个方法的逻辑上。你找到的/numpy/numpy/array_api/_array_object.py里的__add__，就是该Array类为+运算符定义的具体实现。

二、目标__add__方法的执行流程拆解 #

我们逐个拆解代码里的关键步骤：

• other = self._check_allowed_dtypes(other, "numeric", "__add__")：先检查other的类型合法性，必须是数值类型（int/float/Array），不满足就返回NotImplemented，交给Python尝试调用b.__radd__(a)（反向加法逻辑）。
• self, other = self._normalize_two_args(self, other)：这是广播机制的前置核心步骤，会把两个操作数的形状调整为兼容格式——比如把标量转成和数组同维度的虚拟形状，或是把低维数组扩展到和高维数组匹配的维度，确保后续加法符合广播规则。
• res = self._array.__add__(other._array)：调用NumPy原生ndarray的__add__方法，这个方法内置了完整的广播加法逻辑，会按照归一化后的形状完成逐元素计算。
• return self.__class__._new(res)：把底层计算得到的ndarray结果，重新包装成当前的Array对象返回。

三、广播机制的实际运作 #

广播的核心逻辑分两步：

1. 形状兼容检查与归一化：由_normalize_two_args完成，确保两个操作数的形状符合广播规则（比如最后一个维度相等，或是其中一个维度为1）。
2. 逻辑扩展与计算：底层ndarray的__add__会按照广播后的形状，自动完成数组的逻辑扩展（无需实际复制数据），再执行元素级加法。

内容的提问来源于stack exchange，提问作者Attila Vajda