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]])
相关推荐
cxr8282 分钟前
deepseek harness能否指挥Claude code和codex来协同开发
人工智能·智能体
数据智研3 分钟前
【数据分享】中国农产品价格调查年鉴(2004-2025)
大数据·人工智能·数据分析·可视化·数据可视化
阿图灵4 分钟前
OpenCV 图像操作六件套:读取、裁剪、缩放、旋转、通道分割与保存
图像处理·人工智能·python·深度学习·opencv·计算机视觉
小小帅呀5 分钟前
学习 VLA 第 1 天:VLA 概述与学习路线
人工智能
TechEdu2026067 分钟前
[人工智能]Claude(Anthropic):模型能力、智能体与安全工程实践
人工智能·ai
pjj198548 分钟前
opencv-图像透视转换
人工智能·opencv·计算机视觉
aneasystone本尊9 分钟前
学习大模型推理的分词:从文本到 Token
人工智能
浪兎兎10 分钟前
【深度学习】(四)案例:基于PyTorch的全连接神经网络实现手机价格区间预测
pytorch·深度学习·神经网络
牧羊人.33313 分钟前
计算机视觉基础 第 11 章| 特征检测与特征匹配
图像处理·人工智能·opencv·计算机视觉
艾莉丝努力练剑15 分钟前
【AI大模型接入SDK】项目的数据结构设计
数据结构·人工智能·大模型·sdk·文件系统·岗位