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


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