Функция PyTorch torch.unique_consecutive
Справочное руководство Pytorch torch
torch.unique_consecutiveЭто функция в PyTorch, используемая для получения последовательных уникальных значений. Она возвращает смежные последовательные уникальные значения в тензоре.
Определение функции
torch.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)
Примеры использования
Пример
import torch
# Создать тензор
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Получить последовательные уникальные значения
unique_vals = torch.unique_consecutive(x)
print(f"Последовательные уникальные значения: {unique_vals}")
# Возвращает количество
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Количество: {counts}")
# Создать тензор
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Получить последовательные уникальные значения
unique_vals = torch.unique_consecutive(x)
print(f"Последовательные уникальные значения: {unique_vals}")
# Возвращает количество
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Количество: {counts}")
Другие расширения