numpy——广播机制

Tatsukyou / 2023-07-30 / 原文

Numpy的广播机制

广播机制的三大原则:

规则一:如果两个数组的维度不相同,那么小维度的张量的形状将会在最左边补1(添加轴)

规则二:如果两个张量形状在任何一个维度上都不匹配,那么数组的形状会沿着维度为1扩展以匹配另一个张量的shape

规则三:如果两个数组的形状在任何一个维度上都不匹配并且没有任何一个维度为1,那么会引起异常。

注意:广播只会沿着某一维度为1的方向扩展,