torch.kthvalue

torch.kthvalue(input, k, dim=None, keepdim=False, *, out=None)

Возвращает именованную кортеж (values, indices) где values является k элементом по возрастанию в каждой строке input тензора в заданном измерении dim. И indices является индексом каждого найденного элемента.

Если dim не указано, используется последнее измерение input.

Если keepdim равно True, то тензоры values и indices имеют такие же размеры, как input, за исключением измерения dim, где их размер равен 1. В противном случае, dim сжимается (см. torch.squeeze()), в результате чего тензоры values и indices имеют на одно измерение меньше, чем тензор input.

Примечание

Когда input является тензором CUDA и существует несколько допустимых k значений, эта функция может недетерминировано вернуть indices для любого из них.

Параметры:

• input (Tensor) – входной тензор.
• k (int) – k для k-го наименьшего элемента
• dim (int, необязательно) – измерение для поиска k-го значения
• keepdim (bool) – сохраняется ли измерение в выходном тензоре.

Ключевые аргументы:

out (tuple, необязательно) – выходная кортеж из (Tensor, LongTensor) может быть необязательно задан для использования в качестве буферов вывода

Пример:

>>> x = torch.arange(1., 6.)
>>> x
tensor([ 1.,  2.,  3.,  4.,  5.])
>>> torch.kthvalue(x, 4)
torch.return_types.kthvalue(values=tensor(4.), indices=tensor(3))

>>> x=torch.arange(1.,7.).resize_(2,3)
>>> x
tensor([[ 1.,  2.,  3.],
        [ 4.,  5.,  6.]])
>>> torch.kthvalue(x, 2, 0, True)
torch.return_types.kthvalue(values=tensor([[4., 5., 6.]]), indices=tensor([[1, 1, 1]]))