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])
相关推荐
qq_2803914920 分钟前
Java构建飞书消息格式详解
人工智能
七元权24 分钟前
FoundationPose 推理全流程解读:从一帧 RGB-D 到 6D 位姿
论文阅读·深度学习·6d姿态估计
Flynn Liang1 小时前
上线后怎么排查 AI 调用?Spring AI Observability:token/延迟/成本埋点与全链路追踪
java·人工智能·spring·监控·可观测性·observability·springai
测试界的小码农1 小时前
【让 AI 生成的测试用例变成 pytest API 自动化脚本】
人工智能·测试用例·pytest
天天代码码天天2 小时前
我开源了一个家庭备份工具:把手机照片和视频,存回自己的电脑
人工智能
AI源码2 小时前
基于机器学习的古籍文字识别与断句校正系统
人工智能·机器学习
用户8339049606952 小时前
记账大师 AIUI购物记账与交互可靠性实践
人工智能
数智人生局2 小时前
配置驱动的通用 ETL 执行引擎:告别"每个需求写一个 Spark 作业"(改配置不发版)
人工智能
Neurocoda2 小时前
基于 NumPy 从零实现 Tiny-AlexNet 在 MNIST 数据集上的验证
人工智能·深度学习
youcans_2 小时前
【嵌入式软件AI编程】(3)面向 AI 协同开发的 STM32 硬件信息标准化
人工智能·stm32·嵌入式硬件·ai编程