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]])
相关推荐
鬓戈5 分钟前
LLM 训练原理、L20 单卡训练可行性与 nanoGPT 入门
人工智能
xsd202411186 分钟前
微信开源 WeVisDoc:端到端文档解析模型
人工智能
skywalk81638 分钟前
光明记忆(LIMS)v2.3 — 智能记忆管理平台(未开放版本)
人工智能·记忆·光明
弈语道破AI13 分钟前
一稿出片!双十一品牌宣传AI视频生成工具推荐:即梦 Seedance 2.5 全球首发
人工智能·音视频
通信大模型16 分钟前
中国科学 信息科学 | TokenCom: 面向多模态和多任务Token通信的视觉-语言模型
人工智能·计算机视觉·语言模型
LeeZhao@17 分钟前
【具身智能】VLA-DM0.5:让机器人真正理解世界
人工智能·计算机视觉·机器人·agi
kaixin_啊啊22 分钟前
【零基础学AI】第 8 章——让 AI 辅助编程
人工智能
木卫二号Coding40 分钟前
第八十六篇-Strata 部署指南(Ubuntu 22.04 + Tesla V100【单卡】)
linux·运维·人工智能·ubuntu
zpjing~.~41 分钟前
Vim 高效批量替换实战:从 /dev/ 替换为 /dev-ai/
人工智能·编辑器·vim
图特摩斯科技44 分钟前
2026年语义模型、本体和知识图谱面试题自出30道
人工智能·知识图谱·abutiongraph·palantir·ontology·ontoflow