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]])
相关推荐
Deepoch1 分钟前
Deepoc具身模型:让智能轮椅从“避障”转向“预判”
人工智能·科技·开发板·具身模型·deepoc
TopDawn7 分钟前
自然语言处理
人工智能·自然语言处理
种时光的人12 分钟前
Java+AI 无缝衔接:Spring AI 聊天模型入门到精通
java·人工智能·spring·ai·spring ai
Codebee22 分钟前
OoderAgent Apex OS:基于Skills化架构的热插拔启动机制
人工智能
苏打水前端客25 分钟前
【OpenClaw 保姆级教程】第二篇:多渠道接入 + 核心技能上手(附实操案例)
人工智能
何政@25 分钟前
Agent Skills 完全指南:从概念到自定义实践
人工智能·python·大模型·claw·404 not found 罗
码农三叔31 分钟前
(1-2)控制系统基础与人形机器人特点:人形机器人控制的特殊挑战
人工智能·机器学习·机器人·人形机器人
ai产品老杨31 分钟前
源码交付与全协议兼容:企业级 AI 视频中台的二次开发实战
人工智能·音视频
Rick199339 分钟前
Prompt 提示词
人工智能·深度学习·prompt