тензорный поток :: ops :: NthElement

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

Резюме

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

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

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

Аргументы:

• scope: объект Scope
• ввод: 1-D или выше с последним размером не менее n+1 .
• п: 0-D. Положение отсортированного вектора для выбора по последнему измерению (по каждой строке для матриц). Допустимый диапазон 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
)