【Pytorch】torch.bmm()

Hello&＃xff01;
非常感谢您阅读海轰的文章&＃xff0c;倘若文中有错误的地方&＃xff0c;欢迎您指出&＃xff5e;
 
ଘ(੭ˊᵕˋ)੭
昵称&＃xff1a;海轰
标签&＃xff1a;程序猿&＃xff5c;C&＃43;&＃43;选手&＃xff5c;学生
简介&＃xff1a;因C语言结识编程&＃xff0c;随后转入计算机专业&＃xff0c;获得过国家奖学金&＃xff0c;有幸在竞赛中拿过一些国奖、省奖…已保研
学习经验&＃xff1a;扎实基础 &＃43; 多做笔记 &＃43; 多敲代码 &＃43; 多思考 &＃43; 学好英语&＃xff01;
 
唯有努力&＃x1f4aa;
 
本文仅记录自己感兴趣的内容

语法

torch.bmm(input, mat2, *, out&＃61;None) → Tensor

作用

执行存储在 input 和 mat2 中的矩阵的批量矩阵乘积

input 和 mat2 必须是 3-D 张量&＃xff0c;每个张量都包含相同数量的矩阵

如果 input 是 一个$(b\times n\times m)$ 张量&＃xff0c;mat2 是 一个 $(b\times m\times p)$ 张量

out 将是 一个 $(b\times n\times p)$ 张量

此功能不广播。 对于广播矩阵产品&＃xff0c;请参阅 torch.matmul()。

举例

input &＃61; torch.randn(10, 3, 4)
mat2 &＃61; torch.randn(10, 4, 5)
res &＃61; torch.bmm(input, mat2)
print(res.size())


mm,bmm和matmul的区别

参考&＃xff1a;https://blog.csdn.net/qq_40589781/article/details/104223130

参考

• https://pytorch.org/docs/stable/generated/torch.bmm.html#torch.bmm
• https://blog.csdn.net/qq_40589781/article/details/104223130

文章仅作为个人学习笔记记录&＃xff0c;记录从0到1的一个过程

希望对您有一点点帮助&＃xff0c;如有错误欢迎小伙伴指正