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])
相关推荐
武子康1 分钟前
调查研究-222 PagedAttention 详解:vLLM 如何像操作系统一样管理 KV Cache
人工智能·llm·agent
龙虾PRO6 分钟前
AI 产业三大发展主线同步爆发,香港企业依托智能工具落地合规化应用方案
人工智能
B8017913Y12 分钟前
上课记笔记写不完不会整理?2026学习笔记生成使用场景该怎么选
人工智能·笔记·学习
武子康13 分钟前
调查研究-221 KV Cache:LLM 推理服务真正的显存黑洞
人工智能·llm·agent
小猴子下山12316 分钟前
2026年无锡干细胞领域企业发展观察:聚焦细胞生物技术与大健康管理
人工智能·精选
IT_陈寒16 分钟前
React useEffect依赖数组把我坑惨了,这次搞明白了
前端·人工智能·后端
环境栈笔记22 分钟前
AI 浏览器任务失败怎么排查:Context、Session、Page State 和 Task Log 检查清单
前端·人工智能·后端·自动化
Be-real28 分钟前
机器学习:欠拟合、过拟合、偏差方差超全详解
人工智能·机器学习·过拟合·欠拟合
饼干哥哥28 分钟前
Lovart真的帮大忙了,一键直出品牌视觉套件和个人IP封面Skill
人工智能·代码规范·设计
私人珍藏库28 分钟前
[Android] Npatch-免Root框架+可内置模块
android·人工智能·智能手机·工具·软件