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 属性

类成员变量说明

axis

long TopKFusion::axis

描述:

作用的轴

sorted

boolean TopKFusion::sorted

描述:

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