PyTorch torch.triu Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.triuEs ist eine Funktion in PyTorch, die verwendet wird, um den oberen Dreiecksteil einer Matrix zu extrahieren (einschließlich der Hauptdiagonale). Der untere Dreiecksteil wird auf 0 gesetzt.

Funktionsdefinition

torch.triu(input, diagonal=0, out=None)

Parameterbeschreibung:

  • input: Eingabetensor
  • diagonal: Diagonalindex, 0 gibt die Hauptdiagonale an
  • out: Ausgabe-Tensor

Anwendungsbeispiele

Beispiel

import torch

# Eine Matrix erstellen
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Den oberen Dreiecksteil extrahieren
y = torch.triu(a)
print(y)

Das Ausgabeergebnis ist:

tensor([[1, 2, 3],
        [0, 5, 6],
        [0, 0, 9]])

Beispiel

import torch

# Eine Matrix erstellen
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Den oberen Dreiecksteil oberhalb der ersten Diagonale über der Hauptdiagonale extrahieren
y = torch.triu(a, diagonal=1)
print(y)

Das Ausgabeergebnis ist:

tensor([[0, 2, 3],
        [0, 0, 6],
        [0, 0, 0]])

Beispiel

import torch

# Erstelle eine nichtquadratische Matrix
a = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Den oberen Dreiecksteil extrahieren
y = torch.triu(a)
print(y)

Das Ausgabeergebnis ist:

tensor([[1, 2, 3],
        [0, 5, 6]])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen