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-мерный или более высокий с последним измерением не менее n+1. |
n | A Tensor типа int32. 0-мерный. Позиция отсортированного вектора для выбора вдоль последнего измерения (вдоль каждой строки для матриц). Допустимый диапазон n составляет [0, input.shape[:-1]) |
reverse | Необязательный bool. По умолчанию False. Если установлено в True, находит n-е по величине значение в векторе и наоборот. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и 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/r2.3/api_docs/python/tf/raw_ops/NthElement
