PyTorch torch.diagflat Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.diagflatEs ist eine Funktion in PyTorch, mit der eine flache Diagonalmatrix erstellt wird. Unabhängig davon, ob die Eingabe eindimensional oder mehrdimensional ist, wird sie abgeflacht und als Diagonalelemente verwendet.

Funktionsdefinition

torch.diagflat(input, diagonal=0)

Parameterbeschreibung:

  • input: Eingabe-Tensor, wird abgeflacht
  • diagonal: Diagonalindex

Anwendungsbeispiel

Beispiel

import torch

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

# Eine flache Diagonalmatrix erstellen
y = torch.diagflat(x)
print(y)

Das Ausgabeergebnis ist:

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

Beispiel

import torch

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

# Wird zu [1,2,3,4] abgeflacht, wodurch eine 4x4-Diagonalmatrix erstellt wird
y = torch.diagflat(x)
print(y)

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen