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])
相关推荐
程序猿追8 分钟前
棋盘上的博弈:我在 HarmonyOS 里塞了一个五子棋“大脑”
人工智能·华为·harmonyos
是烨笙啊20 分钟前
在 Claude code 中如何利用模型缓存节省 token
人工智能·缓存·ai编程
薛定猫AI1 小时前
【深度解析】从 Claude Mythos 争议看大模型落地:幻觉、Benchmark、成本墙与安全边界
人工智能
混凝土拌意大利面1 小时前
TG-BOOT springboot 功能集散开发框架(AI 协作友好)
人工智能·spring boot·后端
phltxy1 小时前
Spring AI 从提示词到多模态
java·人工智能·spring
标书畅畅行1 小时前
全流程企业级 AI 标书系统技术实现与工程实践
大数据·人工智能
银河麒麟操作系统1 小时前
银河麒麟安全SDK 3.0全面升级
人工智能·安全
赴山海bi1 小时前
AI驱动亚马逊电商增长:DeepBI如何重塑盈利模式
大数据·人工智能
萤丰信息2 小时前
AI赋能安全治理,构筑智慧园区全域智能安防新防线
人工智能·安全
智慧景区与市集主理人2 小时前
市集的 “IP 化” 打造路径——从单次活动到长期品牌资产
人工智能·科技·tcp/ip