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 分钟前
SQL详细语法教程(七)核心优化
数据库·人工智能·sql
居然JuRan6 分钟前
模型压缩“炼金术”:剪枝、量化、知识蒸馏大揭秘
人工智能
初岘17 分钟前
自动驾驶架构:人为接口与隐式特征的博弈
人工智能·架构·自动驾驶
旦莫24 分钟前
MTSC2025参会感悟:AI 驱动的测试用例生成
人工智能·python·测试开发·测试用例·ai测试·mtsc2025
房开民26 分钟前
基于OpenCV的物体识别与计数
人工智能·opencv·计算机视觉
朱程44 分钟前
写给自己的 LangChain 开发教程(一):Hello world & 历史记录
前端·人工智能
有一只柴犬1 小时前
你还在“试错式提问”AI?是时候用上 PromptPilot 了
人工智能·aigc
Wah-Aug1 小时前
深入浅出集成学习:从理论到实战,解锁机器学习 “集体智慧”
人工智能·机器学习·集成学习
Ronin-Lotus1 小时前
图像处理篇---掩膜获取
图像处理·人工智能
小关会打代码1 小时前
计算机视觉第一课opencv(三)保姆级教学
人工智能·opencv·计算机视觉·边缘检测·膨胀·腐蚀