Función torch.unique_consecutive de PyTorch
Manual de referencia de Pytorch torch
torch.unique_consecutiveEs una función de PyTorch utilizada para obtener valores únicos consecutivos. Devuelve los valores únicos consecutivos adyacentes en el tensor.
Definición de la función
torch.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)
Ejemplo de uso
Ejemplo
import torch
# Crear el tensor
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Obtener valores únicos consecutivos
unique_vals = torch.unique_consecutive(x)
print(f"Consecutive unique values: {unique_vals}")
# Devolver el recuento
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
# Crear el tensor
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Obtener valores únicos consecutivos
unique_vals = torch.unique_consecutive(x)
print(f"Consecutive unique values: {unique_vals}")
# Devolver el recuento
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
Otras extensiones