深度学习·pytorch

广播机制

• 从末尾开始逐个维度遍历两个矩阵的shape，如果维度不相同，则考虑广播：
• 任一方的维度为1或者维度不存在 (小矩阵广播为大矩阵)，这样的运算可以广播

可以广播的例子

x=torch.empty(5,3,4,1)
y=torch.empty(3,1,1)
(x.add_(y)).size()

不可以广播的例子

原因：2!=3，违反至少有一个矩阵当前维度为1

x=torch.empty(5,2,4,1)
y=torch.empty(3,1,1)
(x+y).size()

逐元素乘法运算规则