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])
相关推荐
keyanbanyungong22 分钟前
告别杂乱病历!临床科研AI工具实测
人工智能·深度学习
出海小龙27 分钟前
B2B 跟 B2C 的联盟营销有何根本区别?以及分别如何真正推动增长?
大数据·人工智能
xcLeigh30 分钟前
聚合AI工具KULAAI:GPT、Claude、Gemini、DeepSeek热门模型一键使用
人工智能·gpt·claude·gemini·deepseek·聚合ai·kulaai
EnCi Zheng31 分钟前
09aaac-RMSNorm是什么?
人工智能
QiLinkOS32 分钟前
【从实验室到商业战场:发明专利如何重塑科技与企业的共生生态】
大数据·c语言·数据结构·c++·人工智能·单片机·算法
zhangfeng113336 分钟前
AI 日报 | 2026年5月28日
人工智能
ishangy1 小时前
智慧矿山罐笼乘坐人数超限AI检测识别算法
人工智能·ai视觉解决方案·智慧矿山·ai视觉监控·罐笼超员ai检测识别
沐自礼1 小时前
DeepSeekMoE 原理
人工智能·llm
Smartdaili China1 小时前
OpenClaw赋能AI智能体:实时联网与网页抓取
人工智能·爬虫·ai·爬取·openclaw·open claw
汤姆yu1 小时前
SkyClaw‑v1.0昆仑万维天工 AI
大数据·人工智能