在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,此时逐元素相加即可得到结果。
在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,此时逐元素相加即可得到结果。