Fonction torch.addmv de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.addmvC'est une fonction de PyTorch utilisée pour ajouter le résultat d'une multiplication matrice-vecteur au vecteur d'entrée. Elle effectue une multiplication matrice-vecteur, puis ajoute le résultat à l'entrée.

Définition de fonction

torch.addmv(input, mat, vec, *, beta=1.0, alpha=1.0, out=None)

Parameters:

  • input(Tensor) : Le vecteur ou la matrice d'entrée, ajouté au résultat.
  • mat(Tensor) : Matrice d'entrée de forme (n, m).
  • vec(Tensor) : Vecteur d'entrée de forme (m,) ou (m, 1).
  • beta(float, optionnel) : Coefficient multipliant l'entrée, vaut 1.0 par défaut.
  • alpha(float, optionnel) : Coefficient multipliant le résultat de mat @ vec, vaut 1.0 par défaut.
  • out(Tensor, optionnel) : Tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie la somme du résultat de la multiplication matrice-vecteur et du vecteur d'entrée.

Exemple d'utilisation

Exemple

import torch

# Créer le vecteur d'entrée, la matrice et le vecteur
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)

# Effectuer addmv
result = torch.addmv(input, mat, vec)

print("Forme du vecteur d'entrée :", input.shape)
print("Forme de la matrice :", mat.shape)
print("Forme du vecteur :", vec.shape)
print("Forme du résultat :", result.shape)
print(result)

Le résultat de sortie est:

输入向量形状: torch.Size([3])
矩阵形状: torch.Size([3, 4])
向量形状: torch.Size([4])
结果形状: torch.Size([3])
tensor([-1.0182, -0.4901, -0.9490])

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions