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]])
相关推荐
天天讯通3 分钟前
机器人系统对接线索平台好处
大数据·数据库·人工智能·机器人·语音识别
Codebee4 分钟前
低代码与 AI 的深度融合:OneCode3.0 引领企业级应用开发新范式
人工智能·全栈
博睿谷IT99_5 分钟前
2025年华为HCIA-AI认证是否值得考?还是直接冲击HCIP?
人工智能·华为·ai·华为认证·职业规划
Albert_Lsk6 分钟前
【2025/07/31】GitHub 今日热门项目
人工智能·开源·github·开源协议
sali-tec8 分钟前
C# 基于halcon的视觉工作流-章27-带色中线
开发语言·人工智能·算法·计算机视觉·c#
Nuoyunjituan9 分钟前
告别物业思维:科技正重构产业园区的价值坐标系
人工智能·科技·重构
红叶落水30 分钟前
研电赛-基于GD32的纳型无人机AI追踪系统2
人工智能·无人机
棱镜研途35 分钟前
科研快报 |无人机+AI:广东防控基孔热背后的技术革命
图像处理·人工智能·计算机视觉·ai·视觉检测·无人机·基孔肯雅热
斜月42 分钟前
Jupyter Notebook 与 Pandas 绘图实践
人工智能·python
天才少女爱迪生1 小时前
pytorch的自定义 CUDA 扩展怎么学习
人工智能·pytorch·学习