Python mxnet.ndarray.op.argsort用法及代码示例

用法:
mxnet.ndarray.op.argsort(data=None, axis=_Null, is_ascend=_Null, dtype=_Null, out=None, name=None, **kwargs)

参数：

data：(NDArray) - 输入数组
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” 时有效。如果所选数据类型不能精确表示索引，则会引发错误。
out：(NDArray, optional) - 输出 NDArray 来保存结果。

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

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

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

例子：

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.ndarray.op.argsort。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：