PyTorch torch.trace Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.traceist eine Funktion in PyTorch, die verwendet wird, um die Spur einer Matrix zu berechnen. Die Spur ist die Summe der Elemente auf der Hauptdiagonale einer Matrix.

Funktionsdefinition

torch.trace(input)

Parameterbeschreibung:

  • input: Eingabetensor (mindestens zweidimensional)

Anwendungsbeispiel

Beispiel

import torch

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

# Die Spur berechnen (Summe der Elemente der Hauptdiagonale)
y = torch.trace(a)
print(y)

Das Ausgabeergebnis ist:

tensor(15)

Beispiel

import torch

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

# Berechne die Spur
y = torch.trace(a)
print(y)

Das Ausgabeergebnis ist:

tensor(6)

Beispiel

import torch

# Erstelle einen mehrdimensionalen Tensor, berücksichtige nur die ersten beiden Dimensionen
a = torch.randn(3, 4, 4, 5)
y = torch.trace(a)
print(y.shape)

Das Ausgabeergebnis ist:

torch.Size([3, 5])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen