广播机制_torch numpy

参考博文

一些需要记住的关键点
1、何时产生广播机制?(广播机制可以发生的必要条件)
在numpy两个不同shape的数组在进行加减的时候,会发生广播 增加 两个元素在进行乘法的时候也会有广播机制
例子:
在这里插入图片描述

2、如何判断广播是否兼容:
两个计算数组维度相同的时候:需要从后向前一次比零个较各轴的轴长度,如果相同或某轴长度为1,广播兼容可以进行广播机制处理后进行广播计算得到结果;
如果两个数组的维度不同: 那么给低维度的数组前扩展提升一维,扩展维的轴长度为1,然后在扩展出的维上进行广播机制处理。

此外:torch中也有广播机制
加法广播例子

猜你喜欢

转载自blog.csdn.net/m0_45290027/article/details/127200047