Функция PyTorch torch.unique

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.uniqueЭто функция в PyTorch, используемая для получения уникальных значений. Она возвращает уникальные значения в тензоре и может опционально возвращать соответствующие индексы.

Определение функции

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

Примеры использования

Примеры

import torch

# Создайте тензор
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])

# Получить уникальные значения
unique_vals = torch.unique(x)
print(f"Уникальные значения: {unique_vals}")

# Также вернуть индексы
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Обратные индексы: {inverse_idx}")

# Также вернуть количество
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Количество: {counts}")

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения