broadcasting的核心思想:小维度先对齐(小维度在右边),比如a [4,16,16,32],b [32],计算a+b需要对b进行broadcasting,先把b扩展成相等的维度b [1,1,1,32],再扩展成相同的size [4,16,16,32]。
broadcasting好处:1.简洁;2.不需要复制内存,节省空间。
也可进行手动broadcasting。
broadcasting的核心思想:小维度先对齐(小维度在右边),比如a [4,16,16,32],b [32],计算a+b需要对b进行broadcasting,先把b扩展成相等的维度b [1,1,1,32],再扩展成相同的size [4,16,16,32]。
broadcasting好处:1.简洁;2.不需要复制内存,节省空间。
也可进行手动broadcasting。