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])
相关推荐
To_OC6 小时前
搞懂 Token 和 Embedding 后,我终于明白大模型是怎么 "读" 文字的
人工智能·llm·agent
冬奇Lab9 小时前
每日一个开源项目(第139篇):Voicebox - 本地运行的开源 ElevenLabs 替代品
人工智能·开源·资讯
冬奇Lab9 小时前
Skill 系列(03):Skill 设计范式——5 个模式让输出从混沌到可预测
人工智能·开源·agent
IT_陈寒11 小时前
Python搞不定字符串编码?这破玩意坑我两小时!
前端·人工智能·后端
大模型真好玩12 小时前
什么是Loop Engineering?最通俗易懂的Loop Engineering核心概念
人工智能·agent·deepseek
叁两13 小时前
前端转型AI Agent该如何学习?(前置篇)
前端·人工智能·node.js
LaiYoung_13 小时前
🎁 送你一套超好用超实用的 FE AI-Coding Skills
前端·人工智能·开源
ZzT15 小时前
怎么做才不会被 AI 替代?
人工智能·程序员
道友可好15 小时前
从今天开始:你的第一个 Harness Engineering 实践
前端·人工智能·后端