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])
相关推荐
小姜前线技术13 小时前
AI流式渲染打字机效果抖动?节流方案踩坑实录
人工智能
用户0183493016913 小时前
AI对话状态管理:useReducer还是XState
人工智能
先锋部队13 小时前
给AI对话加「停止生成」按钮:abort SSE实战
人工智能
新新技术迷13 小时前
移动端H5接AI对话的坑:键盘顶起与滚动到底
人工智能
aqi0016 小时前
15天学会AI应用开发(七)有了大模型为什么还要引入RAG
人工智能·python·大模型·ai编程·ai应用
用户51914958484517 小时前
libcurl Headers API 释放后重利用漏洞:跨请求复用头句柄导致堆内存安全风险
人工智能·aigc
踩蚂蚁17 小时前
自定义语音唤醒词:从训练到部署的完整链路实践
人工智能
用户51914958484517 小时前
CVE-2025-1094 PostgreSQL SQL注入与WebSocket劫持远程代码执行利用工具
人工智能·aigc
IT_陈寒18 小时前
SpringBoot自动配置这个坑,我踩进去又爬出来了
前端·人工智能·后端