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])
相关推荐
Elastic 中国社区官方博客3 分钟前
用 Elasticsearch 构建一个 ChatGPT connector 来查询 GitHub issues
大数据·人工智能·elasticsearch·搜索引擎·chatgpt·github·全文检索
奔跑的石头_20 分钟前
如何用AI创建一个适合你的编程社区用户名
人工智能
yuhaiqun198920 分钟前
10分钟快速get:零基础AI人工智能学习路线
人工智能·学习
m0_6501082423 分钟前
Co-MTP:面向自动驾驶的多时间融合协同轨迹预测框架
论文阅读·人工智能·自动驾驶·双时间域融合·突破单车感知局限·帧间轨迹预测·异构图transformer
向阳逐梦24 分钟前
电子烟的4种屏幕驱动集成语音方案介绍
人工智能·语音识别
蓝耘智算25 分钟前
蓝耘元生代GPU算力调度云平台深度解析:高性价比算力云与GPU算力租赁首选方案
人工智能·ai·gpu算力·蓝耘
ckjr00727 分钟前
2025 创始人 IP+AI 峰会:见证时代分水岭
人工智能·创客匠人·创客匠人万人峰会
geneculture28 分钟前
2025对2023《融智学导读》升级版,第三章:智能化双字棋盘软件(将智能化双字棋盘定位为第二次认知大飞跃的工作母机是一个极其精准和有力的论断)
人工智能·信息科学·融智学的重要应用·信智序位·全球软件定位系统
ccLianLian34 分钟前
计算机视觉·ZegFormer
人工智能·计算机视觉
i查拉图斯特拉如是38 分钟前
Coze工作流意图识别
人工智能·大模型·工作流·意图识别·coze