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