PyTorch torch.unique_consecutive-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.unique_consecutiveEs handelt sich um eine Funktion in PyTorch, mit der aufeinanderfolgende eindeutige Werte ermittelt werden. Sie gibt benachbarte aufeinanderfolgende eindeutige Werte im Tensor zurück.

Funktionsdefinition

torch.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)

Anwendungsbeispiele

Beispiel

import torch

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

# Aufeinanderfolgende eindeutige Werte abrufen
unique_vals = torch.unique_consecutive(x)
print(f"Aufeinanderfolgende eindeutige Werte: {unique_vals}")

# Anzahl zurückgeben
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Anzahlen: {counts}")

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen