Функция torch.kthvalue в PyTorch
Справочное руководство PyTorch torch
torch.kthvalueЭто функция в PyTorch, используемая для возврата k-го наименьшего значения. Она возвращает k-е наименьшее значение и его индекс вдоль указанного измерения входного тензора.
Определение функции
torch.kthvalue(input, k, dim=None, keepdim=False)
Пример использования
Пример
import torch
# Создать тензор
x = torch.tensor([3, 1, 2, 4])
# Вернуть 2-е наименьшее значение и индекс
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
# Создать тензор
x = torch.tensor([3, 1, 2, 4])
# Вернуть 2-е наименьшее значение и индекс
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
Результат вывода:
tensor(2) tensor(2)
Другие расширения