第6章 运算广播与ufunc

第6章 · 运算广播与ufunc

本章要回答的三个问题：两个数组做运算，内存账本上记了几笔？形状不同的数组为什么能直接相加，广播规则到底怎么执行？ufunc 是什么，out 参数和原地运算能省下多少？

为什么会有这一章

运算是 NumPy 存在的理由，但多数教程只教"加号怎么用"。内存视角下的运算有清晰的账本结构：读两个输入块、写一个输出块，每笔都真金白银。理解账本之后，广播不再是"神奇的对齐"而是一套可执行的形状规则；ufunc 的 out 参数也不再是冷门参数，而是消灭中间数组、把内存占用砍到三分之一的实用工具。

读完能解决什么

• 能对任意二元运算说出输入输出各分配了几块内存
• 能逐步推演广播对齐过程，预判结果形状与报错原因
• 能用 out 参数与原地运算符改写内存紧张的代码
• 能解释 += 与 + 等号重绑定的本质区别

各节怎么分工

本章三节环环相扣：先会记账，才能看懂广播替你省掉的中间块；会用广播，才能理解 out 参数在完整管线里的位置。建议按序通读，不要跳。

本章知识点清单

1. 能写出等形状二元运算的"两读一写"账本，并估算链式表达式的内存峰值
2. 能说出等号只命名不复制，以及复合函数（如 hypot）消灭中间数组的原理
3. 能完整执行广播四步对齐法：右对齐补一、逐维比较、冲突报错、取逐维最大
4. 能用 broadcast_to 与 strides 中的 0 解释"虚拟复制不搬数据"
5. 能用 keepdims 打通"聚合接广播"的形状链条，并排查 (n,1) 与 (1,n) 之错
6. 能说出 ufunc 是元素级运算统一底座，out 与 where 参数全线通用
7. 能解释原地运算"装不下就报错"的类型铁律与整型溢出风险
8. 能用预分配缓冲区改写高频管线，峰值内存恒定

开读前三个疑问

问：广播规则总是记不住，有诀窍吗？
答：别背结论，走流程。四步对齐法每次手工推一遍，推十几遍就内化了。真正要背的只有一句：右对齐、缺维补一、有 1 可扩、其余必须相等。

问：内存账本在什么规模的代码里才值得关心？
答：经验线是"数组超过一百万元素"。小于这个量级临时数组无感；超过之后，每次链式表达式的临时块都是几十上百 MB，账本从理论变成生存技能。

问：原地运算看起来处处更省，为什么不全部用原地？
答：原地销毁原值。调试时想对比中间结果、算法需要回溯历史值、或者原数据还要复用——这些场合原地写入都是灾难。省内存永远排在正确性之后。

先决条件

• 第 5 章的 newaxis 与 squeeze（广播推演的工具）
• 第 3 章的 dtype 提升（运算结果类型由此决定）
• 建议准备一个草稿本：广播推演前期靠手写形状，比纯脑内推演可靠得多，本章三节的实验都值得手推一遍再跑代码验证；三节的代码块都可直接复制运行，改几个数字就能变成自己的测试样例

往下走到哪

运算之后要归纳总结。第 7 章的聚合（沿轴压缩）与线性代数（矩阵乘），是本章元素级运算在两个方向上的延伸：一个做减法（降维），一个做组合（乘加）。本章是全书篇幅最重的一章，三节各有实验，读完回到本页把开篇三问再答一遍，才算过关。