torch.mv

python 复制代码
torch.mv(input, vec, *, out=None) → Tensor

执行矩阵input和向量vec的矩阵向量乘积。

如果input是(n×m)张量,vec是大小为m的1-D张量,out将是大小为n的1-D。

这句话可以理解为:

python 复制代码
如果input是(n×m)张量,vec是大小为m的1-D张量,也就是vec是(mx1)张量,out将是大小为n的1-D,也就是(nx1)的张量,正常的torch.mm。

input(张量)--要相乘的矩阵

vec(张量)--要相乘的矢量

out(张量,可选)--输出张量。

python 复制代码
>>> mat = torch.randn(2, 3)
>>> vec = torch.randn(3)
>>> torch.mv(mat, vec)
tensor([ 1.0404, -0.6361])
相关推荐
yubajin几秒前
【AI】我们没有护城河,OpenAI也没有
人工智能·语言模型
Mr.Daozhi几秒前
从零构建 AI 学术论文助手(四):PDF.js 实时截图 + Gemini 视觉分析
javascript·人工智能·pdf·canvas·pdf.js·gemini
binbin_522 分钟前
React Native 鸿蒙环境搭建完整实战:从 RN 工程到 RNOH 运行
深度学习·react native·react.js·harmonyos
凡科建站8 分钟前
价格便宜的ai建站平台有哪些?2026年价格便宜的ai建站平台推荐
人工智能
黎阳之光9 分钟前
黎阳之光实时三维重构:重构智慧铁路全新管控范式
大数据·人工智能·物联网·算法·数字孪生
Esaka_Forever12 分钟前
动态工具编排:现代 AI Agent 的底层基石 完整深度解释
人工智能
weixin_4000056013 分钟前
Vision-Language-Action:LMDrive项目架构与核心算法组件
人工智能·深度学习·算法·机器学习·架构·自动驾驶
xiezhr15 分钟前
数据库1600张表的文档,我让AI自动维护了
人工智能·ai编程·dba
mingo_敏15 分钟前
强化学习(RL):原理、算法、RLHF落地全解析
人工智能·算法·机器学习