torch.mm

python 复制代码
torch.mm(input, mat2, *, out=None) → Tensor

执行矩阵input和mat2的矩阵乘法运算。

如果input是(n×m)张量,mat2是(m×p)张量,out将是(n x p)张量。

input(张量)--第一个要矩阵相乘的矩阵

mat2(张量)--第二个要矩阵相乘的矩阵

out(张量,可选)--输出张量。

python 复制代码
>>> mat1 = torch.randn(2, 3)
>>> mat2 = torch.randn(3, 3)
>>> torch.mm(mat1, mat2)
tensor([[ 0.4851,  0.5037, -0.3633],
        [-0.0760, -3.6705,  2.4784]])
相关推荐
程序员陆业聪24 分钟前
2025,AI这一年:从“卷参数“到“卷落地“的转折之年
人工智能
DeepModel28 分钟前
第14章 注意力机制与Transformer
人工智能·深度学习·transformer
甲枫叶29 分钟前
【claude】Claude Code正式引入Git Worktree原生支持:Agent全面实现并行独立工作
java·人工智能·git·python·ai编程
ppppppatrick1 小时前
【深度学习基础篇03】激活函数详解:从生物突触到非线性注入
人工智能·深度学习
Figo_Cheung1 小时前
Figo《量子几何学:从希尔伯特空间到全息时空的统一理论体系》(三)
人工智能·深度学习·几何学
人工智能AI技术1 小时前
Qwen3.5-Plus Agent开发实战:从0到1做自动执行AI助手
人工智能
liliangcsdn2 小时前
基于Saliency Map对LLM进行可解释性分析
人工智能·计算机视觉·目标跟踪
盟接之桥2 小时前
盟接之桥说制造:从客供的外在共生到内在的身心合一
运维·服务器·网络·人工智能·制造
RoyLin2 小时前
你的 nginx 在扼杀 AI 服务——为什么需要重新设计流量层
人工智能·devops