PyTorch torch.unique Funktion
PyTorch torch Referenzhandbuch
torch.uniqueEs ist eine Funktion in PyTorch, die verwendet wird, um eindeutige Werte zu erhalten. Sie gibt die eindeutigen Werte im Tensor zurück und kann optional die entsprechenden Indizes zurückgeben.
Funktionsdefinition
torch.unique(input, sorted=True, return_inverse=False, return_counts=False, dim=None)
Anwendungsbeispiele
Beispiele
import torch
# Einen Tensor erstellen
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])
# Eindeutige Werte abrufen
unique_vals = torch.unique(x)
print(f"Eindeutige Werte: {unique_vals}")
# Auch Indizes zurückgeben
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Inverse Indizes: {inverse_idx}")
# Auch Anzahlen zurückgeben
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Anzahlen: {counts}")
# Einen Tensor erstellen
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])
# Eindeutige Werte abrufen
unique_vals = torch.unique(x)
print(f"Eindeutige Werte: {unique_vals}")
# Auch Indizes zurückgeben
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Inverse Indizes: {inverse_idx}")
# Auch Anzahlen zurückgeben
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Anzahlen: {counts}")
Andere Erweiterungen