tf.raw_ops.TopK
Находит значения и индексы k наибольших элементов для последнего измерения.
tf.raw_ops.TopK(
input, k, sorted=True, name=None
)
Если входной параметр является вектором (ранг-1), находит k наибольшие элементы в векторе и выводит их значения и индексы в виде векторов. Таким образом, values[j] является j-ым по величине элементом в input, а его индекс равен indices[j].
Для матриц (соответственно, для входов более высокого ранга) вычисляет k наибольших элементов в каждой строке (соответственно, векторе вдоль последнего измерения). Таким образом,
values.shape = indices.shape = input.shape[:-1] + [k]
Если два элемента равны, элемент с меньшим индексом появляется первым.
Если k изменяется динамически, используйте TopKV2 ниже.
| Аргументы | |
|---|---|
input | A Tensor. Должен быть одним из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. 1-D или более высокой размерности с последним измерением не менее k. |
k | An int который является >= 0. Количество верхних элементов для поиска вдоль последнего измерения (по каждой строке для матриц). |
sorted | Необязательный bool. По умолчанию True. Если значение True, полученные k элементы будут отсортированы по значениям в порядке убывания. |
name | Название операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж из Tensor объектов (значения, индексы). | |
values | A Tensor. Имеет тот же тип, что и input. |
indices | A Tensor типа int32. |
© 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/r2.4/api_docs/python/tf/raw_ops/TopK
