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原生重构医疗新范式|博鳌超级AI医院亮相消博会,悦尔APP+SupMed超超解锁全民智慧健康
人工智能·重构·健康医疗·ai-native
A_QXBlms6 分钟前
企微工具对比:用API打通AI私域客服自动回复(实测3款工具)
人工智能·企业微信
wenzhangli720 分钟前
从零开始的 SPAC 编程构建 BPM 设计器实战
人工智能
踩着两条虫20 分钟前
效率翻倍!AI智能体深度解析:自然语言 → DSL → Vue组件
前端·人工智能·低代码
小江的记录本25 分钟前
【Transformer架构】Transformer架构核心知识体系(包括自注意力机制、多头注意力、Encoder-Decoder结构)
java·人工智能·后端·python·深度学习·架构·transformer
AI先驱体验官26 分钟前
债小白分析:债务优化服务的新变量、AI能否带来行业升级
大数据·人工智能·深度学习·重构·aigc
青苔猿猿29 分钟前
OpenWebUI(20)源码学习-版本升级
人工智能·学习·ai·openwebui
黑金IT29 分钟前
当博弈论遇见AI开发,研发即博弈,对抗即进化
人工智能·自动化
ai生成式引擎优化技术31 分钟前
服装店AI生成式引擎优化(GEO)方案
人工智能
mseaspring33 分钟前
离线语音识别 Vosk 入门指南:开源、轻量、告别网络依赖
人工智能·语音识别