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

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/cc/class/tensorflow/ops/nth-element