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]])
相关推荐
今日说"法"几秒前
线性代数与矩阵运算:向量、矩阵、特征值、SVD 在 AI 中的全面应用
人工智能·线性代数·矩阵
前端技术几秒前
[特殊字符]️ Spring AI Alibaba Advisor基础应用
java·人工智能·spring
实在智能RPA3 分钟前
电商运营自动化AI工具有哪些?哪个最好用?——2026年全链路智能体选型深度指南
大数据·人工智能·ai·自动化
刘佬GEO5 分钟前
没时间写内容还能做 GEO:方法、流程与可操作方案
大数据·网络·人工智能·搜索引擎·ai
星速云5 分钟前
开源AI工具生态全景:20+工具如何对接统一API网关
人工智能·gpt·开源·api·claude
今日说"法"5 分钟前
线性代数与矩阵运算:AI 背后的数学基石
人工智能·线性代数·矩阵
X journey5 分钟前
机器学习实践(18.5):特征工程补充
人工智能·算法·机器学习
北鹤M12 分钟前
2026实测:17款降重降AI工具横评降重鸟登顶
人工智能
IT_陈寒15 分钟前
Python的GIL把我CPU跑满时我才明白并发不是这样玩的
前端·人工智能·后端
wanghowie16 分钟前
21.AI运营助手整体架构设计:Multi-Agent + 语义层 + 数据系统
人工智能