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部署优化技术分析:从云端到边缘的智能化演进
人工智能·gpt·aigc·边缘计算·ai编程·ai写作·agi
chushiyunen2 分钟前
大模型.safetensors文件
人工智能·pytorch·深度学习
带娃的IT创业者2 分钟前
信号链双路径陷阱:新增 Signal 路径后 AI 回复重复的根因与修复
人工智能
光羽隹衡3 分钟前
计算机视觉——Opencv(人脸检测)
人工智能·opencv·计算机视觉
IT_陈寒5 分钟前
SpringBoot 项目启动慢?5 个提速技巧让你的应用快如闪电 ⚡️
前端·人工智能·后端
币之互联万物5 分钟前
好用的推理训练引擎:博云AIOS如何重塑企业AI算力底座
大数据·人工智能
视***间7 分钟前
算力下沉,智赋全域——解析当前AI边缘计算形势及视程空间的使命与征程
人工智能·边缘计算·ai算力·视程空间·终端算力
IT_陈寒7 分钟前
SpringBoot自动配置的坑,我把头发都快薅没了
前端·人工智能·后端
supericeice8 分钟前
建筑行业大模型落地提速,知识图谱方案为何受到关注
人工智能·知识图谱
小橙子学AI8 分钟前
OpenClaw 定时任务与提醒:打造你的 AI 自动化工作流
运维·人工智能·自动化