tf.contrib.nn.nth_element
Находит значения n-го по величине элемента для последнего измерения.
tf.contrib.nn.nth_element(
input, n, reverse=False, name=None
)
Обратите внимание, что n имеет нулевую индексацию.
Если вход представляет собой вектор (ранг-1), находит элементы, которые являются n-м по величине значением в векторе, и выводит их значения в виде скалярного тензора.
Для матриц (соответственно, входных данных более высокого ранга) вычисляет элементы, которые являются n-м по величине значением в каждой строке (соответственно, векторе по последнему измерению). Таким образом,
values.shape = input.shape[:-1]
| Аргументы | |
|---|---|
input | 1-D или более высокий Tensor с последним измерением как минимум n+1. |
n | Tensor тип int32. 0-D. Позиция отсортированного вектора для выбора по последнему измерению (по каждой строке для матриц). Допустимый диапазон n составляет [0, input.shape[:-1]) |
reverse | Необязательный bool. По умолчанию False. При установке в True находит n-е по величине значение в векторе и наоборот. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Tensor. Имеет тот же тип, что и input. n-й порядковый статистический параметр по каждому срезу последнего измерения. |
© 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/r1.15/api_docs/python/tf/contrib/nn/nth_element
