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