Fonction torch.unique_consecutive de PyTorch
Manuel de référence Pytorch torch
torch.unique_consecutiveIl s'agit d'une fonction de PyTorch utilisée pour obtenir des valeurs uniques consécutives. Elle renvoie les valeurs uniques consécutives adjacentes dans le tenseur.
Définition de fonction
torch.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)
Exemples d'utilisation
Exemple
import torch
# Créer un tenseur
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Obtenir des valeurs uniques consécutives
unique_vals = torch.unique_consecutive(x)
print(f"Valeurs uniques consécutives : {unique_vals}")
# Renvoyer les décomptes
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
# Créer un tenseur
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Obtenir des valeurs uniques consécutives
unique_vals = torch.unique_consecutive(x)
print(f"Valeurs uniques consécutives : {unique_vals}")
# Renvoyer les décomptes
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
Autres extensions