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
: Then
-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::
|
Optional attribute setters for NthElement . |
Public attributes
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