Pytorch广播机制

在b站看李沐的视频，讲到广播机制的时候，有点晕晕的。

做下学习笔记：

 生成两个张量，求他们的和：

均为二维矩阵，a是3*1，b是1*2，由于维度不同不能逐元素相加 ，因此需要广播机制。

将a和b变为更大的矩阵：取最大的行数和列数，即便为3*2的矩阵

处理张量a：将a由3*1变为3*2， 复制第一列作为第二列： 

 

处理张量b：将b由1*2，变为3*2，复制第一行作为第三第四行：

 得到新的a和b张量，两个张量均为3*2，此时逐元素相加即可得到结果。