pytorch:torch.mm()和torch.matmul()的使用

脚本专栏 2024/11/6 佚名

2 0 1

神剑山庄资源网 Design By www.hcban.com

如下所示：

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()的使用就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。

pytorch,torch.mm,torch.matmul

标签：

神剑山庄资源网 Design By www.hcban.com

神剑山庄资源网 免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com

神剑山庄资源网 Design By www.hcban.com

评论“pytorch:torch.mm()和torch.matmul()的使用”

暂无pytorch:torch.mm()和torch.matmul()的使用的评论...

2024年11月06日