tf.keras.ops.sort

Sorts the elements of x along a given axis in ascending order.

x Input tensor.
axis Axis along which to sort. If None, the tensor is flattened before sorting. Defaults to -1; the last axis.

Sorted tensor.