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])
相关推荐
NAGNIP1 小时前
轻松搞懂全连接神经网络结构!
人工智能·算法·面试
moshuying2 小时前
别让AI焦虑,偷走你本该有的底气
前端·人工智能
董董灿是个攻城狮3 小时前
零基础带你用 AI 搞定命令行
人工智能
喝拿铁写前端5 小时前
Dify 构建 FE 工作流:前端团队可复用 AI 工作流实战
前端·人工智能
阿里云大数据AI技术6 小时前
阿里云 EMR Serverless Spark + DataWorks 技术实践:引领企业 Data+AI 一体化转型
人工智能
billhan20166 小时前
MCP 深入理解:协议原理与自定义开发
人工智能
Jahzo6 小时前
openclaw桌面端体验--ClawX
人工智能·github
billhan20166 小时前
Agent 开发全流程:从概念到生产
人工智能
用户1474853079746 小时前
AI-动手深度学习环境搭建-d2l
深度学习