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]]))