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]

Args:

  • 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.

Constructors and Destructors

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)

Public attributes

operation
values

Public functions

node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const

Public static functions

Reverse(bool x)

Structs

tensorflow::ops::NthElement::Attrs

Optional attribute setters for NthElement.

Public attributes

operation

Operation operation

values

::tensorflow::Output values

Public functions

NthElement

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

NthElement

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

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const 

Public static functions

Reverse

Attrs Reverse(
  bool x
)