tf.raw_ops.NthElement

Находит значения n-го порядкового статистического показателя для последнего измерения.

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.raw_ops.NthElement

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`.

© 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