Fonction torch.mv de PyTorch

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

torch.mvest une fonction de PyTorch utilisée pour effectuer une multiplication matrice-vecteur. Elle calcule le produit d'une matrice et d'un vecteur.

Définition de fonction

torch.mv(input, vec, out=None)

Parameters:

  • input(Tensor) : matrice d'entrée, forme (n, m).
  • vec(Tensor) : vecteur d'entrée, forme (m,) ou (m, 1).
  • out(Tensor, optionnel) : Tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie le produit matrice-vecteur, forme (n,).

Exemple d'utilisation

Exemple

import torch

# Créer une matrice et un vecteur
mat = torch.randn(3, 4)
vec = torch.randn(4)

# Multiplication matrice-vecteur
result = torch.mv(mat, vec)

print("Forme de la matrice :", mat.shape)
print("Forme du vecteur :", vec.shape)
print("Forme du résultat :", result.shape)
print("Résultat :", result)

Le résultat de sortie est:

矩阵形状: torch.Size([3, 4])
向量形状: torch.Size([4])
结果形状: torch.Size([3])
tensor([-0.1861,  0.2482, -0.4375])

Exemple - Calcul d'une couche de réseau de neurones

import torch

# Simuler une couche entièrement connectée d'un réseau de neurones
W = torch.randn(512, 256)  # Matrice de poids
x = torch.randn(256)        # Vecteur d'entrée

# Calculer y = W @ x
y = torch.mv(W, x)

print("Forme de la matrice de poids :", W.shape)
print("Forme du vecteur d'entrée :", x.shape)
print("Forme du vecteur de sortie :", y.shape)

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

Autres extensions