Python mxnet.ndarray.op.topk用法及代碼示例

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

參數：

data：(NDArray) - 輸入數組
axis：(int or None, optional, default='-1') - 選擇前 k 個索引的軸。如果未給出，則使用展平數組。默認值為 -1。
k：(int, optional, default='1') - 要選擇的頂部元素數，應始終小於或等於給定軸中的元素數。如果設置 k < 1，則執行全局排序。
ret_typ：({'both', 'indices', 'mask', 'value'},optional, default='indices') - 返回類型。 “value”表示返回前k個值，“indices”表示返回前k個值的索引，“mask”表示返回一個包含0和1的掩碼數組。1表示前k個值。 “both” 表示返回前 k 個元素的值和索引的列表。
is_ascend：(boolean, optional, default=0) - 是選擇 k 個最大的還是 k 個最小的元素。如果設置為 false，將選擇前 K 個最大的元素。
dtype：({'float16', 'float32', 'float64', 'int32', 'int64', 'uint8'},optional, default='float32') - ret_typ 為 “indices” or “both” 時的輸出索引的 DType。如果所選數據類型不能精確表示索引，則會引發錯誤。
out：(NDArray, optional) - 輸出 NDArray 來保存結果。

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

返回輸入數組中頂部 k 元素的索引: 軸(默認)。如果 ret_type 設置為 ‘value’，則返回頂部 k 元素的值(而不是索引)。在 ret_type = ‘both’ 的情況下，將返回值和索引。返回的元素將被排序。

例子：

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

// returns an index of the largest element on last axis
topk(x) = [[ 2.],
           [ 1.]]

// returns the value of top-2 largest elements on last axis
topk(x, ret_typ='value', k=2) = [[ 0.4,  0.3],
                                 [ 0.3,  0.2]]

// returns the value of top-2 smallest elements on last axis
topk(x, ret_typ='value', k=2, is_ascend=1) = [[ 0.2 ,  0.3],
                                             [ 0.1 ,  0.2]]

// returns the value of top-2 largest elements on axis 0
topk(x, axis=0, ret_typ='value', k=2) = [[ 0.3,  0.3,  0.4],
                                         [ 0.1,  0.2,  0.2]]

// flattens and then returns list of both values and indices
topk(x, ret_typ='both', k=2) = [[[ 0.4,  0.3], [ 0.3,  0.2]] ,  [[ 2.,  0.], [ 1.,  2.]]]

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.op.topk。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：