Función PyTorch torch.unique

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.uniqueEs una función de PyTorch utilizada para obtener valores únicos. Devuelve los valores únicos del tensor y, opcionalmente, puede devolver los índices correspondientes.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Crear tensor
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])

# Obtener valores únicos
unique_vals = torch.unique(x)
print(f"Unique values: {unique_vals}")

# Devolver también los índices
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Inverse indices: {inverse_idx}")

# Devolver también los recuentos
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Counts: {counts}")

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones