PyTorch torch.addmv 函数
Pytorch torch 参考手册torch.addmv 是 PyTorch 中用于将矩阵向量乘法的结果添加到输入向量的函数。它执行矩阵与向量的乘法，然后将结果与 input 相加。
函数定义
torch.addmv(input, mat, vec, *, beta=1.0, alpha=1.0, out=None)
参数:
input(Tensor): 输入向量或矩阵，被添加到结果中。mat(Tensor): 输入矩阵，形状为 (n, m)。vec(Tensor): 输入向量，形状为 (m,) 或 (m, 1)。beta(float, 可选): 乘以 input 的系数，默认为 1.0。alpha(float, 可选): 乘以 mat @ vec 结果的系数，默认为 1.0。out(Tensor, 可选): 输出张量。
返回值:
torch.Tensor: 返回矩阵向量乘法结果与输入向量之和。
使用示例
实例
import torch
# 创建输入向量、矩阵和向量
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)
# 执行 addmv
result = torch.addmv(input, mat, vec)
print("输入向量形状:", input.shape)
print("矩阵形状:", mat.shape)
print("向量形状:", vec.shape)
print("结果形状:", result.shape)
print(result)
# 创建输入向量、矩阵和向量
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)
# 执行 addmv
result = torch.addmv(input, mat, vec)
print("输入向量形状:", input.shape)
print("矩阵形状:", mat.shape)
print("向量形状:", vec.shape)
print("结果形状:", result.shape)
print(result)
输出结果为：
输入向量形状: torch.Size([3]) 矩阵形状: torch.Size([3, 4]) 向量形状: torch.Size([4]) 结果形状: torch.Size([3]) tensor([-1.0182, -0.4901, -0.9490])