PyTorch torch.addbmm-Funktion
PyTorch torch-Referenzhandbuch
torch.addbmmEs ist eine Funktion in PyTorch, die verwendet wird, um die Ergebnisse der Batch-Matrixmultiplikation zu einer Eingabematrix zu addieren. Sie führt eine Matrixmultiplikation für jedes Matrizenpaar in batch1 und batch2 durch, summiert alle Ergebnisse und addiert sie dann zu input.
Funktionsdefinition
torch.addbmm(input, batch1, batch2, *, beta=1.0, alpha=1.0, out=None)
Parameters:
input(Tensor): Die Eingabematrix, die zum Ergebnis addiert wird.batch1(Tensor): Die erste Batch-Matrix mit der Form (b, n, m).batch2(Tensor): Die zweite Batch-Matrix mit der Form (b, m, p).beta(float, optional): Koeffizient, mit input multipliziert, Standardwert ist 1.0.alpha(float, optional): Koeffizient, mit dem Ergebnis von batch1 @ batch2 multipliziert, Standardwert ist 1.0.out(Tensor, optional): Der Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt die Summe aus dem Ergebnis der Batch-Matrixmultiplikation und der Eingabematrix zurück.
Anwendungsbeispiel
Beispiel
import torch
# Eingabematrix und Batch-Matrizen erstellen
input = torch.randn(3, 3)
batch1 = torch.randn(10, 3, 4)
batch2 = torch.randn(10, 4, 3)
# addbmm ausführen
result = torch.addbmm(input, batch1, batch2)
print("Form der Eingabematrix:", input.shape)
print("Form der Batch-Matrix 1:", batch1.shape)
print("Form der Batch-Matrix 2:", batch2.shape)
print("Form des Ergebnisses:", result.shape)
print(result)
# Eingabematrix und Batch-Matrizen erstellen
input = torch.randn(3, 3)
batch1 = torch.randn(10, 3, 4)
batch2 = torch.randn(10, 4, 3)
# addbmm ausführen
result = torch.addbmm(input, batch1, batch2)
print("Form der Eingabematrix:", input.shape)
print("Form der Batch-Matrix 1:", batch1.shape)
print("Form der Batch-Matrix 2:", batch2.shape)
print("Form des Ergebnisses:", result.shape)
print(result)
Das Ausgabeergebnis ist:
输入矩阵形状: torch.Size([3, 3])
批量矩阵1形状: torch.Size([10, 3, 4])
批量矩阵2形状: torch.Size([10, 4, 3])
结果形状: torch.Size([3, 3])
tensor([[-0.2875, -0.6518, 0.1207],
[-0.2746, 0.5563, -0.2281],
[ 0.3315, 0.1649, 0.2145]])
Andere Erweiterungen