广播 (broadcasting)
飞桨的广播机制主要遵循如下规则(参考 ):
每个张量至少为一维张量
例如:
现在我们知道什么情况下两个张量是可以广播的,两个张量进行广播语义后的结果张量的形状计算规则如下:
保证两个张量形状相等之后,每个维度上的结果维度就是当前维度上较大的那个。
例如:
x = paddle.ones((2, 1, 4))
y = paddle.ones((3, 1))
# z的形状: [2,3,4]
y = paddle.ones((3, 2))
# ValueError: (InvalidArgument) Broadcast dimension mismatch.