TopK

Находит значения и индексы k крупнейших элементов для последнего измерения.

Если входные данные являются вектором (ранг-1), находит k крупнейших записей в векторе и выводит их значения и индексы в виде векторов. Таким образом, «values[j]» — это «j»-я по величине запись в «input», а ее индекс — «indices[j]».

Для матриц (соответственно входные данные более высокого ранга) вычисляет верхние элементы `k` в каждой строке (соответственно вектор по последнему измерению). Таким образом,

значения.shape = index.shape = input.shape[:-1] + [k]

Если два элемента равны, первым отображается элемент с меньшим индексом.