tf.sort
Сортирует тензор.
tf.sort(
values, axis=-1, direction='ASCENDING', name=None
)
Использование:
import tensorflow as tf
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
b = tf.sort(a,axis=-1,direction='ASCENDING',name=None)
c = tf.keras.backend.eval(b)
# Here, c = [ 1. 2.8 10. 26.9 62.3 166.32]
| Аргументы |
values | Численный тензор размерности 1 или выше. Tensor |
axis | Ось, по которой выполняется сортировка. По умолчанию -1, что соответствует сортировке по последней оси. |
direction | Направление сортировки значений ('ASCENDING' или 'DESCENDING'). |
name | Необязательное имя операции. |
| Возвращаемое значение |
Тензор с тем же типом данных и формой, что и values, с элементами, отсортированными по заданной axis. |
| Исключения |
ValueError | Если ось не является скалярной константой или направление некорректно. |