harmony 鸿蒙TopKFusion

  • 2023-06-24
  • 浏览 (578)

TopKFusion

概述

查找沿axis轴的前K个最大值和对应索引。

该算子对应的NodeType为NODE_TYPE_TOPK_FUSION。

输入:

  • x,n维tensor。

输出:

  • output0,axis维度的前K个最大值。

  • output1,axis维度的前K个最大值的索引。

Since:

3.2

Version:

1.0

相关模块:

NNRt

汇总

Public 属性

名称 描述
sorted boolean
如果为True,按照大到小排序,如果为False,按照小到大排序。 
axis long
作用的轴 

类成员变量说明

axis

long TopKFusion::axis

描述:

作用的轴

sorted

boolean TopKFusion::sorted

描述:

如果为True,按照大到小排序,如果为False,按照小到大排序。

你可能感兴趣的鸿蒙文章

harmony 鸿蒙HDI 参数参考

harmony 鸿蒙ActRecognitionEvent

harmony 鸿蒙Activation

harmony 鸿蒙ActivityRecognitionTypes.idl

harmony 鸿蒙AddFusion

harmony 鸿蒙Alignment

harmony 鸿蒙AllocInfo

harmony 鸿蒙ApduSimIORequestInfo

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙Audio

0  赞