Python – método tensorflow.argsort()

TensorFlow es una biblioteca de Python de código abierto diseñada por Google para desarrollar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo. Tensorflow tiene un método argsort() que se usa para encontrar los índices de un tensor en orden.

Sintaxis: tf.argsort(valores, eje, dirección, estable, nombre)

Parámetros:

  • valores: Es un Tensor numérico de cualquier dimensión.
  • eje: Define el eje a lo largo del cual se debe realizar el cortocircuito. Si no se proporciona ningún valor, el valor predeterminado es -1 y la clasificación se realiza en función del último eje.
  • dirección : ASCENDENTE o DESCENDENTE .
  • estable: Verdadero  o Falso . Si es cierto entonces en caso de igualdad se mantiene el orden original.
  • nombre: Es un argumento opcional que define el nombre de la operación.

Retorno: Devuelve un Tensor de tipo int32 que tiene la misma forma que los valores. Este tensor contiene los índices que darán el orden de los valores dados.

Si el eje o la dirección no son válidos, generará ValueError.

Ejemplo 1:

Python3

# importing the library
import tensorflow
  
# initializing value
a= [1,5,2.5,10,7,8.5]
  
# getting the indices for sorted values
b = tensorflow.argsort(a,axis=-1,direction='ASCENDING',stable=False)
  
# printing the result
print('Indices:'b)
  
print('Sorted values')
#printing the sorted value
for i in b:
  print(a[i])

Producción:

Indices: tf.Tensor([0 2 1 4 5 3], shape=(6,), dtype=int32)
Sorted Values
1
2.5
5
7
8.5
10

Ejemplo 2: En este ejemplo, se pasa un valor incorrecto a la dirección. Esto generará ValueError.

Python3

# importing the library
import tensorflow
  
# initializing value
a= [1,5,2.5,10,7,8.5]
  
# getting the indices for sorted values
b = tensorflow.argsort(a,axis=-1,direction='ASC',stable=False)

Producción:

ValueError: ASC should be one of ASCENDING, DESCENDING

Publicación traducida automáticamente

Artículo escrito por aman neekhara y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *