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

用法:

参数：

返回：

返回类型：