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