参考博文
一些需要记住的关键点
1、何时产生广播机制?(广播机制可以发生的必要条件)
在numpy两个不同shape的数组在进行加减的时候,会发生广播 增加 两个元素在进行乘法的时候也会有广播机制
例子:
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/fb8b097f790c4f251daef8b2d3621763.png)
2、如何判断广播是否兼容:
两个计算数组维度相同的时候:需要从后向前一次比零个较各轴的轴长度,如果相同或某轴长度为1,广播兼容可以进行广播机制处理后进行广播计算得到结果;
如果两个数组的维度不同: 那么给低维度的数组前扩展提升一维,扩展维的轴长度为1,然后在扩展出的维上进行广播机制处理。
此外:torch中也有广播机制
加法广播例子