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]])
相关推荐
我没胡说八道12 小时前
论文排版避坑指南|按问题选工具,不踩坑,选好工具
人工智能·深度学习·考研·计算机视觉·自然语言处理·论文
天天讯通12 小时前
AI辅助如何打造呼叫中心的“金牌坐席”
人工智能
wanzehongsheng12 小时前
追日光伏花的双轴追踪机构与控制逻辑拆解
人工智能·光伏发电·光伏·绿色能源·低碳环保
拥抱太阳061612 小时前
HarmonyOS 应用开发《掌上英语》第6篇-资源管理最佳实践多模块资源复用
pytorch·深度学习·harmonyos
带娃的IT创业者12 小时前
当推理速度突破物理极限:深度解析 MiMo-v2.5-Pro-UltraSpeed 的 1000 TPS 架构革命
人工智能·架构·大模型·架构优化·mimo·tps·推理速度
kiros_wang12 小时前
Osaurus 智能生成效果与能力边界实测
人工智能
Lottie202612 小时前
2026跨境铺货破局!1688/淘宝货源全自动采集对接技术
大数据·人工智能
大模型真好玩12 小时前
LangChain DeepAgents 速通指南(十一)—— DeepAgents Code 记忆与状态管理
人工智能·langchain·agent
weixin_3975740912 小时前
企业级AI框架的五大核心能力
人工智能·ai
weipt12 小时前
MCP协议在企业中的实际应用
人工智能