Python mxnet.ndarray.argsort用法及代碼示例

用法:
mxnet.ndarray.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.argsort。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：