tf.keras.ops.top_k

Находит k наибольших значений и их индексы в тензоре.

tf.keras.ops.top_k(
    x, k, sorted=True
)

Аргументы
`x`	Входной тензор.
`k`	Целое число, представляющее количество наибольших элементов для извлечения.
`sorted`	Логическое значение, указывающее, нужно ли сортировать результат в порядке убывания. По умолчанию `True`.

Возвращаемое значение
Кортеж из двух тензоров. Первый тензор содержит k наибольших значений, а второй — индексы этих значений во входном тензоре.

Пример:

x = keras.ops.convert_to_tensor([5, 2, 7, 1, 9, 3])
values, indices = top_k(x, k=3)
print(values)
array([9 7 5], shape=(3,), dtype=int32)
print(indices)
array([4 2 0], shape=(3,), dtype=int32)

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/top_k