tf.math.top_k

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

Просмотреть псевдонимы

Основные псевдонимы

tf.nn.top_k

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.math.top_k, tf.compat.v1.nn.top_k, `tf.compat.v2.math.top_k`, `tf.compat.v2.nn.top_k`

tf.math.top_k(
    input, k=1, sorted=True, name=None
)

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

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

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

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

Аргументы
`input`	1-мерный или многомерный `Tensor` с последним измерением не менее `k`.
`k`	0-мерный `int32` `Tensor`. Количество верхних элементов для поиска по последнему измерению (по каждой строке для матриц).
`sorted`	Если True, результирующие `k` элементы будут отсортированы по значениям в порядке убывания.
`name`	Необязательное имя операции.

Возвращаемые значения
`values`	`k` самых больших элементов вдоль каждого слайса последнего измерения.
`indices`	Индексы `values` в пределах последнего измерения `input`.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/math/top_k