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])
相关推荐
骄阳如火2 分钟前
论文撰写SKILLS实测二|academic-research-skills:带“反幻觉内核“的研究→写作→评审全流水线
人工智能
办公室马主任3 分钟前
华南机械加工企业选MES服务商怎么选?
大数据·运维·人工智能·制造
冬奇Lab12 分钟前
代码库知识库系列(10):增量更新——什么时候该重建索引,重建哪些部分
人工智能
碳基猿25 分钟前
新媒体矩阵运营API是什么?企业如何通过API打造自动化内容分发系统?
人工智能·新媒体运营·新媒体矩阵·运营数据统计·矩阵分发
≮傷£≯√31 分钟前
opencv 图片合并
人工智能·opencv·计算机视觉
冬奇Lab41 分钟前
开源项目第180期:Omnigent — Databricks 出品的 AI Agent 元编排层,让 Claude Code、Codex、Cursor 统一管控
人工智能·开源·agent
kattgatt1 小时前
卡特加特玄武大模型合规备案,商用更安心
人工智能·卡特加特
qq_348231851 小时前
如何使用 Codex
人工智能
IT_陈寒1 小时前
Java并行流把我坑惨了:原来不是线程安全的!
前端·人工智能·后端
石逸凡1 小时前
AI驱动的金融IT架构转型升级
人工智能·金融·架构