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])
その他の拡張