PyTorch torch.diag Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.diagIst eine Funktion in PyTorch, die verwendet wird, um diagonale Matrizen zu erstellen oder diagonale Elemente eines Tensors zu extrahieren.

Funktionsdefinition

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

Parameterbeschreibung:

  • input: Eingabe-Tensor
  • diagonal: Diagonalindex, 0 steht für die Hauptdiagonale, positive Werte stehen für obere Diagonalen, negative Werte für untere Diagonalen

Anwendungsbeispiel

Beispiel

import torch

# Einen eindimensionalen Tensor erstellen
x = torch.tensor([1, 2, 3])

# Eine diagonale Matrix erstellen
y = torch.diag(x)
print(y)

Das Ausgabeergebnis ist:

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

Beispiel

import torch

# Diagonale Elemente aus einer Matrix extrahieren
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Die Hauptdiagonale extrahieren
y = torch.diag(x)
print(y)

Das Ausgabeergebnis ist:

tensor([1, 5, 9])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen