tf.raw_ops.TopKV2

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

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

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

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

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

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