tensorflow::ops::NthElement

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

Краткое содержание

Если на входе находится вектор (ранг 1), функция находит элементы, являющиеся n-ми наименьшими значениями в векторе, и выводит их значения в виде скалярного тензора.

Для матриц (соответственно, входных данных более высокого ранга) вычисляется элемент, представляющий собой n-е наименьшее значение в каждой строке (соответственно, векторе вдоль последнего измерения). Таким образом,

values.shape = input.shape[:-1]

Аргументы:

• scope: Объект Scope
• Входные данные: одномерные или многомерные, причем последнее измерение должно быть не менее n+1 .
• n: 0-мерный. Позиция отсортированного вектора для выбора вдоль последнего измерения (вдоль каждой строки для матриц). Допустимый диапазон n: [0, input.shape[:-1])

Дополнительные атрибуты (см. Attrs ):

• reverse: Если установлено значение True, находит n-е по величине значение в векторе, и наоборот.

Возвращает:

• Output : n -я порядковая статистика вдоль каждого последнего среза измерения.

Общедоступные атрибуты

Operation operation

::tensorflow::Output values

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n
)

 NthElement(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input n,
  const NthElement::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs Reverse(
  bool x
)