tf.raw_ops.NthElement | TensorFlow v2.16.1 (original) (raw)
tf.raw_ops.NthElement
Stay organized with collections Save and categorize content based on your preferences.
Finds values of the n-th order statistic for the last dimension.
View aliases
Compat aliases for migration
SeeMigration guide for more details.
tf.compat.v1.raw_ops.NthElement
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. |