pytorch元素相乘_pytorch 中矩阵乘法总结

本文介绍了PyTorch中处理矩阵乘法的不同方法,包括二维矩阵乘法torch.mm(),三维带batch的矩阵乘法torch.bmm(),多维矩阵乘法torch.matmul(),以及逐元素乘法torch.mul(),并详细解释了各函数的工作原理和适用场景。
摘要由CSDN通过智能技术生成

源于多维tensor的乘法不知道应该应用哪个函数,然后发现了别人总结的好,特意借鉴过来,以备不时之需。

1. 二维矩阵乘法 torch.mm()

torch.mm(mat1, mat2, out=None)

其中

equation?tex=mat1+%5Cin+%5Cmathbb%7BR%7D%5E%7Bn%5Ctimes+m%7D ,

equation?tex=mat2+%5Cin+%5Cmathbb%7BR%7D%5E%7Bm+%5Ctimes+d%7D , 输出的

equation?tex=out+%5Cin+%5Cmathbb%7BR%7D%5E%7Bn+%5Ctimes+d%7D

该函数一般只用来计算两个二维矩阵的矩阵乘法,并且不支持broadcast操作。

2. 三维带batch的矩阵乘法 torch.bmm()

由于神经网络训练一般采用mini-batch,经常输入的时三维带batch的矩阵,所以提供

torch.bmm(bmat1, bmat2,

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值