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


Python mxnet.symbol.argsort用法及代码示例


用法:

mxnet.symbol.argsort(data=None, axis=_Null, is_ascend=_Null, dtype=_Null, name=None, attr=None, out=None, **kwargs)

参数

  • data(Symbol) - 输入数组
  • axis(int or None, optional, default='-1') - 对输入张量进行排序的轴。如果未给出,则使用展平数组。默认值为 -1。
  • is_ascend(boolean, optional, default=1) - 是升序还是降序排序。
  • dtype({'float16', 'float32', 'float64', 'int32', 'int64', 'uint8'},optional, default='float32') - 输出索引的 DType。仅当ret_typ 为“indices” or “both” 时有效。如果所选数据类型不能精确表示索引,则会引发错误。
  • name(string, optional.) - 结果符号的名称。

返回

结果符号。

返回类型

Symbol

返回将沿给定轴对输入数组进行排序的索引。

此函数沿给定轴执行排序,并返回一个索引数组,其形状与输入数组的形状相同,该输入数组按排序顺序索引数据。

例子:

x = [[ 0.3,  0.2,  0.4],
     [ 0.1,  0.3,  0.2]]

// sort along axis -1
argsort(x) = [[ 1.,  0.,  2.],
              [ 0.,  2.,  1.]]

// sort along axis 0
argsort(x, axis=0) = [[ 1.,  0.,  1.]
                      [ 0.,  1.,  0.]]

// flatten and then sort
argsort(x, axis=None) = [ 3.,  1.,  5.,  0.,  4.,  2.]

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.symbol.argsort。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。