tf.raw_ops.TopKV2

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

tf.raw_ops.TopKV2(
    input, k, sorted=True, name=None
)

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

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

values.shape = indices.shape = input.shape[:-1] + [k]

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