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.
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.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.
[null,null,["Last updated 2024-04-26 UTC."],[],[],null,["# tf.raw_ops.NthElement\n\n\u003cbr /\u003e\n\nFinds values of the `n`-th order statistic for the last dimension.\n\n#### View aliases\n\n\n**Compat aliases for migration**\n\nSee\n[Migration guide](https://www.tensorflow.org/guide/migrate) for\nmore details.\n\n[`tf.compat.v1.raw_ops.NthElement`](https://www.tensorflow.org/api_docs/python/tf/raw_ops/NthElement)\n\n\u003cbr /\u003e\n\n tf.raw_ops.NthElement(\n input, n, reverse=False, name=None\n )\n\nIf the input is a vector (rank-1), finds the entries which is the nth-smallest\nvalue in the vector and outputs their values as scalar tensor.\n\nFor matrices (resp. higher rank input), computes the entries which is the\nnth-smallest value in each row (resp. vector along the last dimension). Thus, \n\n values.shape = input.shape[:-1]\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|-----------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `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| `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])` |\n| `reverse` | An optional `bool`. Defaults to `False`. When set to True, find the nth-largest value in the vector and vice versa. |\n| `name` | A name for the operation (optional). |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A `Tensor`. Has the same type as `input`. ||\n\n\u003cbr /\u003e"]]
