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]])
相关推荐
searchforAI几秒前
CC-Switch教程:统一管理Skills、MCP、模型供应商、系统提示词等多项配置
人工智能·gpt·ai·大模型·agent·claudecode
tedcloud1233 分钟前
codegraph部署教程:构建代码库语义分析环境
服务器·人工智能·word·excel
城事漫游Molly4 分钟前
AI赋能质性研究(四):AI辅助写分析备忘录
人工智能·prompt·ai for science·定性研究
北京软秦科技有限公司7 分钟前
AI报告审核加持,IACheck助力企业数智化转型与检测报告质量再造
人工智能
道可云10 分钟前
智造有道:深度解读《山东省“人工智能+制造”行动方案(2026—2028年)》
人工智能·制造
长空任鸟飞_阿康13 分钟前
驾驭 AI 这匹野马:深入解析智能体 Harness 工程
人工智能·python·ai
YOLO数据集集合20 分钟前
智慧电网红外热成像数据集|电力设备组件识别目标检测深度学习数据集
人工智能·深度学习·yolo·目标检测·计算机视觉
2601_9507607920 分钟前
CyQuantiFluor™细胞活力检测试剂盒检测原理详解
人工智能·蛋白
老A的AI实验室25 分钟前
Cyber Weekly #66
人工智能·ai·llm·agi·genai
Ada's28 分钟前
【LLM基础研究】核心七:PPO
人工智能