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]])
相关推荐
大锅盖12 分钟前
HarmonyOS 6.1.1 AI字幕新特性深度解析:从AICaptionComponent四大新增字段到K歌社交舞台黑霓虹紫深色架构全链路实战
人工智能·架构·harmonyos
烟锁池塘柳02 分钟前
课程学习(Curriculum Learning, CL):一种模仿人类从易到难学习过程的机器学习训练策略
人工智能·机器学习
麻雀飞吧11 分钟前
近期量化工具怎么选,先看你卡在哪一环
人工智能·python
java1234_小锋12 分钟前
YOLO26 计算机视觉 - YOLO26 简介 & Hello World项目搭建
人工智能·yolo·计算机视觉·机器视觉·yolo26
阿图灵12 分钟前
OpenCV 图像特征与匹配:SIFT 特征检测与 BFMatcher 暴力匹配
图像处理·人工智能·python·opencv·计算机视觉·sift
Mr数据杨16 分钟前
商店商品销量预测助力库存优化
人工智能·数据分析·kaggle竞赛
东方佑19 分钟前
202 个参数干掉 59 万个参数:ConvLinear 的设计、融合与实战
人工智能
唯创知音21 分钟前
电动车仪表语音芯片方案 速度、电量、故障三语播报用WT588F02-8S-C
人工智能·语音识别·电动车仪表语音芯片方案
前沿在线22 分钟前
Vbot ATOM发布,人形机器人进入产品化时刻
人工智能·ai·大模型