Finds values of the n
-th order statistic for the last dimension.
tf.raw_ops.NthElement(
input, n, reverse=False, name=None
)
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 | |
---|---|
input
|
A Tensor . Must be one of the following types: float32 , float64 , int32 , uint8 , int16 , int8 , int64 , bfloat16 , uint16 , half , uint32 , uint64 .
1-D or higher with last dimension at least n+1 .
|
n
|
A Tensor of type int32 .
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])
|
reverse
|
An optional bool . Defaults to False .
When set to True, find the nth-largest value in the vector and vice
versa.
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as input .
|