PyTorch torch.addmv Funktion
PyTorch torch-Referenzhandbuch
torch.addmvEs ist eine Funktion in PyTorch, die verwendet wird, um das Ergebnis der Matrix-Vektor-Multiplikation zum Eingabevektor zu addieren. Sie führt eine Matrix-Vektor-Multiplikation durch und addiert dann das Ergebnis zur Eingabe.
Funktionsdefinition
torch.addmv(input, mat, vec, *, beta=1.0, alpha=1.0, out=None)
Parameters:
input(Tensor): Der Eingabevektor oder die Eingabematrix, der/die zum Ergebnis addiert wird.mat(Tensor): Eingabematrix mit Form (n, m).vec(Tensor): Eingabevektor mit Form (m,) oder (m, 1).beta(float, optional): Koeffizient, mit dem die Eingabe multipliziert wird; Standardwert ist 1.0.alpha(float, optional): Koeffizient, mit dem das Ergebnis von mat @ vec multipliziert wird; Standardwert ist 1.0.out(Tensor, optional): Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt die Summe aus dem Ergebnis der Matrix-Vektor-Multiplikation und dem Eingabevektor zurück.
Anwendungsbeispiel
Beispiel
import torch
# Eingabevektor, Matrix und Vektor erstellen
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)
# addmv ausführen
result = torch.addmv(input, mat, vec)
print("Form des Eingabevektors:", input.shape)
print("Form der Matrix:", mat.shape)
print("Form des Vektors:", vec.shape)
print("Form des Ergebnisses:", result.shape)
print(result)
# Eingabevektor, Matrix und Vektor erstellen
input = torch.randn(3)
mat = torch.randn(3, 4)
vec = torch.randn(4)
# addmv ausführen
result = torch.addmv(input, mat, vec)
print("Form des Eingabevektors:", input.shape)
print("Form der Matrix:", mat.shape)
print("Form des Vektors:", vec.shape)
print("Form des Ergebnisses:", result.shape)
print(result)
Das Ausgabeergebnis ist:
输入向量形状: torch.Size([3]) 矩阵形状: torch.Size([3, 4]) 向量形状: torch.Size([4]) 结果形状: torch.Size([3]) tensor([-1.0182, -0.4901, -0.9490])
Andere Erweiterungen