Numpy & Pytorch Broadcast机制

矩阵/Tensor之间的叉乘( matmul / @ ):

矩阵乘法规则:

( N , M ) @ ( M , P ) = ( N , P )

如果两个矩阵/Tensor的形状相同,则两个矩阵/Tensor的点乘*的结果就是两个矩阵/Tensor对应位置相乘,要求矩阵/Tensor维数相同,且各个维度的长度相同。

当运算中的两个矩阵/Tensor的形状不同时,Numpy和Pytorch将自动触发broadcast机制。

Broadcast机制:两个矩阵broadcast之后的结果每一维都是两个矩阵中对应维度的最大值,如果维数不一致则在前面加一补齐。

使用broadcast必须满足两个规则:相对应的维数相等或有一个矩阵的维数是1。

最后结果的计算方法是:先将两个矩阵都broadcast到结果的维数,然后再按照相同维度的矩阵对应元素相乘、相加。

矩阵和向量之间的矩阵乘法:

在左边的向量是行向量,在右边的向量是列向量,这样的好处是结果矩阵一定是一个向量。

矩阵和向量之间的按元素乘法、加法:

 

 

 

 

 

 

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Think@

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值