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

© 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/r2.4/api_docs/python/tf/raw_ops/NthElement