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]])
相关推荐
小橙子学AI2 分钟前
Rokid AI眼镜皮肤健康检测智能体技术实践
人工智能
QQ676580083 分钟前
智慧工地物料堆积识别 工地钢筋木材图像识别 工地砖块目标检测 建筑物大理石图像识别 建筑物工地材料识别 物料堆积识别10349期
人工智能·目标检测·计算机视觉·工地物料堆积·工地钢筋木材图像识别·工地砖块目标检测·建筑物大理石图像
懂AI的老郑6 分钟前
时空智能体技术研究与应用
人工智能
Wild API8 分钟前
多模型成本治理怎么落地?从任务分层、日志统计到结构优化的一套实战思路
大数据·网络·人工智能
前端摸鱼匠12 分钟前
【AI大模型春招面试题22】层归一化(Layer Norm)与批归一化(Batch Norm)的区别?为何大模型更倾向于使用Layer Norm?
开发语言·人工智能·面试·求职招聘·batch
人工智能AI技术16 分钟前
索引基础:数据库索引结构与优化原则
人工智能
永霖光电_UVLED19 分钟前
像“黏土”一样被光塑造的材料
大数据·人工智能·汽车·制造·娱乐
逻辑驱动的ken20 分钟前
Java高频面试考点场景题10
java·开发语言·深度学习·求职招聘·春招
wechat_Neal20 分钟前
新能源整车配电方案解析
人工智能·汽车
前端摸鱼匠22 分钟前
【AI大模型春招面试题23】大模型的参数量、计算量如何计算?FLOPs与FLOPS的区别?
开发语言·人工智能·面试·求职招聘·batch