PyTorch torch.tril Funktion
PyTorch torch-Referenzhandbuch
torch.trilist eine Funktion in PyTorch, die verwendet wird, um den unteren Dreiecksteil einer Matrix zu extrahieren (einschließlich der Hauptdiagonale). Der obere Dreiecksteil wird auf 0 gesetzt.
Funktionsdefinition
torch.tril(input, diagonal=0, out=None)
Parameterbeschreibung:
input: Eingabetensordiagonal: Diagonalindex, 0 gibt die Hauptdiagonale anout: Ausgabe-Tensor
Anwendungsbeispiel
Beispiel
import torch
# Eine Matrix erstellen
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# Unteren Dreiecksteil extrahieren
y = torch.tril(a)
print(y)
# Eine Matrix erstellen
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# Unteren Dreiecksteil extrahieren
y = torch.tril(a)
print(y)
Die Ausgabe ist:
tensor([[1, 0, 0],
[4, 5, 0],
[7, 8, 9]])
Beispiel
import torch
# Eine Matrix erstellen
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# Unteren Dreiecksteil extrahieren, beginnend mit der ersten Diagonale unterhalb der Hauptdiagonale
y = torch.tril(a, diagonal=1)
print(y)
# Eine Matrix erstellen
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# Unteren Dreiecksteil extrahieren, beginnend mit der ersten Diagonale unterhalb der Hauptdiagonale
y = torch.tril(a, diagonal=1)
print(y)
Die Ausgabe ist:
tensor([[1, 2, 0],
[4, 5, 6],
[7, 8, 9]])
Beispiel
import torch
# Erstelle eine nichtquadratische Matrix
a = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Unteren Dreiecksteil extrahieren
y = torch.tril(a)
print(y)
# Erstelle eine nichtquadratische Matrix
a = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Unteren Dreiecksteil extrahieren
y = torch.tril(a)
print(y)
Die Ausgabe ist:
tensor([[1, 0, 0],
[4, 5, 0]])
Andere Erweiterungen