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])
相关推荐
Frag0ut几秒前
Chrome DevTools 2026进阶:AI辅助调试与性能诊断原生实战指南
前端·人工智能·chrome devtools·ai调试·lighthouse性能分析·内存泄漏诊断·css容器查询
林伽一3 分钟前
GPU 调度栈开源与框架层精炼:AI 基础设施的两个新落点|2026年10月20日
人工智能·科技·安全·ai·开源
武子康4 分钟前
从预约改期看 Agent 工具粒度:把不变量留在权威服务
人工智能·后端·agent
Wanda12148 分钟前
机器人与机电一体化3D数字孪生机器-Day5
人工智能·unity·游戏引擎·数字孪生
frankzch11 分钟前
HybridAgent 拆解:在潜空间想,动手时才出字
人工智能·深度学习·语言模型
Martina_032112 分钟前
AI生成模型怎么做网页3D轻量化?用6步压缩Mesh、贴图与材质调用
图像处理·人工智能·3d·自然语言处理·aigc·游戏策划
张彦峰ZYF13 分钟前
用一个业务场景画出生产级 Agent 的全景架构
人工智能·生产级 agent·模型、工具、记忆·企业任务执行 agent·接入、会话与上下文·编排、模型访问、工具与集成·数据与状态、治理与观测
小李同学_LHY14 分钟前
大白话模型第三章:深度学习
人工智能·深度学习
Maynor99616 分钟前
TypeSafe Jev 1.14.0 使用教程:System One 决策 API 从入门到 box 定位
人工智能·api·jev·system one·决策模型·typesafe·jev-1.14.0