Sorts a tensor.
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.sort(
values, axis=-1, direction='ASCENDING', name=None
)
Used in the notebooks
Used in the tutorials |
---|
Usage:
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
tf.sort(a).numpy()
array([ 1. , 2.8 , 10. , 26.9 , 62.3 , 166.32], dtype=float32)
tf.sort(a, direction='DESCENDING').numpy()
array([166.32, 62.3 , 26.9 , 10. , 2.8 , 1. ], dtype=float32)
For multidimensional inputs you can control which axis the sort is applied
along. The default axis=-1
sorts the innermost axis.
mat = [[3,2,1],
[2,1,3],
[1,3,2]]
tf.sort(mat, axis=-1).numpy()
array([[1, 2, 3],
[1, 2, 3],
[1, 2, 3]], dtype=int32)
tf.sort(mat, axis=0).numpy()
array([[1, 1, 1],
[2, 2, 2],
[3, 3, 3]], dtype=int32)
See also | |
---|---|
|
Returns | |
---|---|
A Tensor with the same dtype and shape as values , with the elements
sorted along the given axis .
|
Raises | |
---|---|
tf.errors.InvalidArgumentError
|
If the values.dtype is not a float or
int type.
|
ValueError
|
If axis is not a constant scalar, or the direction is invalid. |