tensorflow::ops::NthElement

#include <nn_ops.h>

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

Summary

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

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

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

Аргументы:

• scope: Объект Scope
• input: 1-мерный или многомерный тензор с последним измерением не менее 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
)