tensorflow::ops::NthElement

#include <nn_ops.h>

Finds values of the n-th order statistic for the last dimension.

Summary

If the input is a vector (rank-1), finds the entries which is the nth-smallest value in the vector and outputs their values as scalar tensor.

For matrices (resp. higher rank input), computes the entries which is the nth-smallest value in each row (resp. vector along the last dimension). Thus,

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

Arguments:

• scope: A Scope object
• input: 1-D or higher with last dimension at least n+1.
• n: 0-D. Position of sorted vector to select along the last dimension (along each row for matrices). Valid range of n is [0, input.shape[:-1])

Optional attributes (see Attrs):

• reverse: When set to True, find the nth-largest value in the vector and vice versa.

Returns:

• Output: The n-th order statistic along each last dimensional slice.

Public attributes

::tensorflow::Output values

Public functions

 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 

Public static functions

Attrs Reverse(
  bool x
)