Fonction torch.unique de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.uniqueC'est une fonction dans PyTorch utilisée pour obtenir des valeurs uniques. Elle renvoie les valeurs uniques dans le tenseur et peut éventuellement renvoyer les indices correspondants.

Définition de fonction

torch.unique(input, sorted=True, return_inverse=False, return_counts=False, dim=None)

Exemples d'utilisation

Exemples

import torch

# Créer un tenseur
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])

# Obtenir des valeurs uniques
unique_vals = torch.unique(x)
print(f"Unique values: {unique_vals}")

# Renvoyer également les indices
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Inverse indices: {inverse_idx}")

# Renvoyer également les comptages
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Counts: {counts}")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions