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-D или выше Tensor. |
axis | Ось, по которой выполняется сортировка. По умолчанию -1, что сортирует последнюю ось. |
direction | Направление сортировки значений ('ASCENDING' или 'DESCENDING'). |
name | Необязательное имя операции. |
| Возвращаемое значение |
Тензор Tensor с тем же типом данных и формой, что и values, с элементами, отсортированными вдоль заданной axis. |
| Исключения |
ValueError | Если ось не является константой или направление некорректно. |