tf.keras.ops.argsort

Возвращает индексы, которые бы отсортировали тензор.

Просмотр псевдонимов

Основные псевдонимы

tf.keras.ops.numpy.argsort

tf.keras.ops.argsort(
    x, axis=-1
)

Аргументы
`x`	Входной тензор.
`axis`	Ось, по которой происходит сортировка. По умолчанию `-1` (последняя ось). Если `None`, используется сплющенный тензор.

Возвращаемое значение
Тензор индексов, которые сортируют `x` вдоль указанной `axis`.

Примеры:

Одномерный массив:

x = keras.ops.array([3, 1, 2])
keras.ops.argsort(x)
array([1, 2, 0], dtype=int32)

Двумерный массив:

>>> x = keras.ops.array([[0, 3], [3, 2], [4, 5]])
>>> x
array([[0, 3],
       [3, 2],
       [4, 5]], dtype=int32)
>>> keras.ops.argsort(x, axis=0)
array([[0, 1],
       [1, 0],
       [2, 2]], dtype=int32)
>>> keras.ops.argsort(x, axis=1)
array([[0, 1],
       [1, 0],
       [0, 1]], dtype=int32)

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/argsort