📜  Python – tensorflow.argsort() 方法

📅  最后修改于: 2022-05-13 01:55:37.854000             🧑  作者: Mango

Python – tensorflow.argsort() 方法

TensorFlow 是由 Google 设计的开源Python库,用于开发机器学习模型和深度学习神经网络。 Tensorflow 有一个方法 argsort() 用于按排序顺序查找张量的索引。

示例 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])


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)


输出:

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

示例 2:在此示例中,错误的值是传递给方向。这将引发 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)

输出:

ValueError: ASC should be one of ASCENDING, DESCENDING