Функция sort

Функция sort возвращает тензор с отсортированными элементами вдоль заданной оси. Первым параметром функция принимает исходный тензор. Вторым параметром можно передать номер оси (по умолчанию -1). Третьим параметром задается направление сортировки: direction='ASCENDING' для возрастания или direction='DESCENDING' для убывания.

Синтаксис

tf.sort(values, axis=-1, direction='ASCENDING', name=None)

Пример

Давайте отсортируем одномерный тензор по возрастанию:

import tensorflow as tf

t = tf.constant([3, 1, 5, 2, 4])
res = tf.sort(t)
print(res)

Результат выполнения кода:

tf.Tensor([1 2 3 4 5], shape=(5,), dtype=int32)

Пример

Давайте отсортируем тензор по убыванию, указав параметр direction:

<+python+> import tensorflow as tf t = tf.constant([3, 1, 5, 2, 4]) res = tf.sort(t, direction='DESCENDING') print(res) <-python+>

Результат выполнения кода:

tf.Tensor([5 4 3 2 1], shape=(5,), dtype=int32)

Пример

Давайте отсортируем двумерный тензор вдоль строк (ось 1):

import tensorflow as tf

t = tf.constant([[3, 1, 2], [6, 4, 5]])
res = tf.sort(t, axis=1)
print(res)

Результат выполнения кода:

tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)

Пример

Давайте отсортируем двумерный тензор вдоль столбцов (ось 0):

import tensorflow as tf

t = tf.constant([[3, 1, 2], [6, 4, 5]])
res = tf.sort(t, axis=0)
print(res)

Результат выполнения кода:

tf.Tensor(
[[3 1 2]
 [6 4 5]], shape=(2, 3), dtype=int32)

Смотрите также

функцию argsort,
которая возвращает индексы отсортированных элементов
функцию top_k,
которая возвращает k наибольших элементов
функцию reduce_max,
которая находит максимальное значение тензора
функцию reduce_min,
которая находит минимальное значение тензора