-
什么是广播机制
广播是python用于科学计算的拓展库NumPy在对不同shape的array进行按位运算时,如果参与计算的数组shape不一样,会在一定规则的前提下,将较小的array“广播”成更大的、可以计算的array,否则则会报错。
比如:矩阵A1*n + b(标量) ,在实际操作时会首先将 b 广播成 1*n 的vector 然后才进行 加 操作。广播类似一种在一定规则下会自动对原来数值进行维度上的扩充操作。
-
具体规则
通用规则:
当在两个数组上操作时,NumPy比较它们的每一个维度。它从后面的维度开始,然后向前推进。当出现如下情况:
1、他们是维度相同
2、他们当中有一个维度为1
就看作这两个维度是相容的。如果不满足上述条件的程序则会报错“ValueError: frames are not aligned。
当满足上述条件后:输出数组(计算后的结果)的维度是每一个维度的最大值,广播将值为1的维度进行“复制”、“拉伸”。
网络上的其他资源提到的规则:
1、让所有输入数组都向其中shape最长的数组看齐,shape中不足的部分都通过在前面加1补齐
2、输出数组的shape是输入数组shape的各个轴上的最大值
3、如果输入数组的某个轴和输出数组的对应轴的长度相同或者其长度为1时,这个数组能够用来计算,否则出错
4、当输入数组的某个轴的长度为1时,沿着此轴运算时都用此