当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tensorflow.argsort()用法及代码示例


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

用法:tf.argsort(values, axis, direction, stable, name)

参数:

  • values:它是任何尺寸的数字张量。
  • axis:它定义了需要进行短路的轴。如果未指定任何值,则默认值为-1,并基于最后一个轴进行排序。
  • direction:升序或降序。
  • stable:是非题。如果是真的,则在相等的情况下会保持原始顺序。
  • name:这是一个可选参数,用于定义操作的名称。

返回:它返回一个int32类型的张量,其形状与值相同。该张量包含索引,这些索引将给出给定值的排序顺序。

如果轴或方向无效,则会引发ValueError。

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

输出:

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





注:本文由纯净天空筛选整理自 Python – tensorflow.argsort() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。