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 分钟前
微软AI核电计划
大数据·人工智能·语言模型·自然语言处理
老A的AI实验室11 分钟前
AI Agent的20个趋势洞察
人工智能·ai·语言模型·llm·genai
scott19851226 分钟前
Nerf与3D Gaussian splatting 数学原理与理解
人工智能·深度学习
Skrrapper44 分钟前
智谱AI:CogVideoX-2b——视频生成模型的得力工具
人工智能·音视频
橙子小哥的代码世界1 小时前
深度学习03-神经网络01-什么是神经网络?
人工智能·深度学习·神经网络
小R资源1 小时前
【AI视频】AI虚拟主播制作网站推荐
人工智能·ai数字人·ai虚拟主播·免费ai视频生成·免费ai生成虚拟主播
红米煮粥1 小时前
深度学习-神经网络构造
人工智能·深度学习·神经网络
蜡笔小新星1 小时前
计算机视觉学习路线
人工智能·经验分享·学习·计算机视觉
FL16238631291 小时前
[数据集][目标检测]棉花叶子病害检测数据集VOC+YOLO格式977张22类别
人工智能·yolo·目标检测
爱笑的源码基地2 小时前
Java+Spring Cloud +UniApp 智慧工地源码,用户PC端、移动端数据同步,支持多端展示
人工智能·spring cloud·源码·软件开发·智慧工地·智慧工地app·住建数据监管