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]])
相关推荐
AI-小柒5 分钟前
Seedance 2.0(即梦 2.0)深度解析:AI 视频进入「导演级」可控时代
大数据·人工智能·网络协议·tcp/ip·http·音视频
ZhengEnCi9 分钟前
07. 向量数据库构建与优化
人工智能
lynn-fish15 分钟前
标讯 “加速度”:AI 如何重构电力招投标的决策与效率
人工智能·ai·电网·电力·招投标·标讯
TImCheng060916 分钟前
落地实施方案:为销售团队引入AI工具的四阶段(选型-试点-培训-复盘)攻略
大数据·人工智能
进击monkey18 分钟前
告别Confluence卡顿和权限迷宫:PandaWiki三个月实操体验
人工智能
仙女修炼史26 分钟前
Making Convolutional Networks Shift-Invariant Again
人工智能·深度学习·计算机视觉
EasyGBS28 分钟前
从“联网互通”到“安全可信”:EasyGBS支持GB35114国密协议,覆盖全场景安防合规升级
大数据·人工智能·安全·gb28181·gb35114
程序员Linc32 分钟前
小智esp32小玩意2:了解面包板与小智最基本AI对话电路连接
人工智能
奔跑草-32 分钟前
【AI日报】每日AI最新消息2026-02-24
人工智能