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])
相关推荐
Shockang8 小时前
AI 设计工作流全景拆解:Figma MCP / Claude Design / Codex / Google Stitch
人工智能
To_OC9 小时前
数据集划分不是随便切:手把手切分大众点评情感数据集
人工智能·llm·agent
冬奇Lab10 小时前
每日一个开源项目(第142篇):android/skills - Google 官方 Android 开发 AI Skill 库
人工智能·开源·资讯
冬奇Lab10 小时前
Skill 系列(06):Skill 工程化与治理——路由准确率 38%、压缩节省 76%
人工智能·开源·agent
IT_陈寒12 小时前
Vue这个坑我跳了两次,原来问题出在这
前端·人工智能·后端
weiwei2284412 小时前
神经网络模型导出及开放标准格式ONNX
pytorch·onnx
新新技术迷13 小时前
Node给AI接口做SSE代理与鉴权
人工智能
redreamSo13 小时前
大模型是不是到顶了?瓶颈到底在哪
人工智能·openai
Oo92013 小时前
Tool Use 背后的技术逻辑
人工智能