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])
相关推荐
霍小毛3 分钟前
轻量黑科技!数字孪生重构智慧城乡供水一体化新范式
人工智能
如此风景6 分钟前
Playwright 速查表
人工智能
草青工作室6 分钟前
AI主流大模型参数量和收费情况参考(26年4月)
人工智能
Y敲键盘的地方9 分钟前
第1章:从命令行到智能体
人工智能
雷工笔记15 分钟前
读书笔记|《AI知识库:个人与企业的智慧玩法》
人工智能
李可以量化24 分钟前
【2026 量化工具选型】通达信 TdxQuant vs 迅投 QMT/miniQMT 深度对比:新手该怎么选?
大数据·人工智能·区块链·通达信·qmt·量化 qmt ptrade
互联科技报33 分钟前
零售数字化:高准确率客流分析系统优质推荐
大数据·人工智能
互联科技报34 分钟前
成熟零售客流系统该怎么选?决定门店效率的关键选择
人工智能·零售
北京耐用通信40 分钟前
国产优选:耐达讯自动化EtherCAT转RS232在工业协议转换中的卓越表现
人工智能·科技·物联网·网络协议·自动化
沃垠AI1 小时前
万字干货!Agent Skills从入门到精通
人工智能