PyTorch torch.inner Funktion
PyTorch torch-Referenzhandbuch
torch.innerEs ist eine Funktion in PyTorch, die verwendet wird, um das innere Produkt zweier Tensoren zu berechnen. Für Vektoren ist sie äquivalent zum Skalarprodukt; für höherdimensionale Tensoren berechnet sie das innere Produkt entlang angegebener Dimensionen.
Funktionsdefinition
torch.inner(input, other, out=None)
Parameters:
input(Tensor): Der erste Eingabe-Tensor.other(Tensor): Der zweite Eingabetensor.out(Tensor, optional): Der Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt das Ergebnis des inneren Produkts zurück.
Anwendungsbeispiele
Beispiel - Inneres Produkt von Vektoren
import torch
# Zwei Vektoren erstellen
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([4.0, 5.0, 6.0])
# Berechne das innere Produkt
result = torch.inner(a, b)
print("Vektor a:", a)
print("Vektor b:", b)
print("Ergebnis des inneren Produkts:", result)
# Zwei Vektoren erstellen
a = torch.tensor([1.0, 2.0, 3.0])
b = torch.tensor([4.0, 5.0, 6.0])
# Berechne das innere Produkt
result = torch.inner(a, b)
print("Vektor a:", a)
print("Vektor b:", b)
print("Ergebnis des inneren Produkts:", result)
Das Ausgabeergebnis ist:
向量 a: tensor([1., 2., 3.]) 向量 b: tensor([4., 5., 6.]) 内积结果: tensor(32.)
Beispiel - Inneres Produkt von Matrizen
import torch
# Zwei Matrizen erstellen
A = torch.randn(2, 3)
B = torch.randn(2, 3)
# Das innere Produkt entlang der letzten Dimension berechnen
result = torch.inner(A, B)
print("A-Form:", A.shape)
print("B-Form:", B.shape)
print("Form des Ergebnisses:", result.shape)
# Zwei Matrizen erstellen
A = torch.randn(2, 3)
B = torch.randn(2, 3)
# Das innere Produkt entlang der letzten Dimension berechnen
result = torch.inner(A, B)
print("A-Form:", A.shape)
print("B-Form:", B.shape)
print("Form des Ergebnisses:", result.shape)
Andere Erweiterungen