pytorch:torch.mm()和torch.matmul()的使用
更新时间:2019年12月27日 15:18:23 作者:大雄没有叮当猫
今天小编就为大家分享一篇pytorch:torch.mm()和torch.matmul()的使用,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
如下所示:
torch.mm(mat1, mat2, out=None) → Tensor torch.matmul(mat1, mat2, out=None) → Tensor
对矩阵mat1和mat2进行相乘。 如果mat1 是一个n×m张量,mat2 是一个 m×p 张量,将会输出一个 n×p 张量out。
参数 :
mat1 (Tensor) – 第一个相乘矩阵
mat2 (Tensor) – 第二个相乘矩阵
out (Tensor, optional) – 输出张量
代码示例:
import torch a=torch.randn(2,3) b=torch.randn(3,2) print(torch.mm(a,b)) print(torch.matmul(a,b))
以上这篇pytorch:torch.mm()和torch.matmul()的使用就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Django bulk_create()、update()与数据库事务的效率对比分析
这篇文章主要介绍了Django bulk_create()、update()与数据库事务的效率对比分析,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
在Python3中使用asyncio库进行快速数据抓取的教程
这篇文章主要介绍了在Python3中使用asyncio进行快速数据抓取,asyncio是一个异步IO库,运行效率较高,需要的朋友可以参考下2015-04-04


最新评论