- 接口功能:返回输入Tensor在指定维度上的k个极值及索引。
每个算子分为,必须先调用"aclnnTopkGetWorkspaceSize"接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用"aclnnTopk"接口执行计算。
[object Object]
[object Object]
- 确定性计算:
- aclnnTopk默认确定性实现。
[object Object]
每个算子分为,必须先调用"aclnnTopkGetWorkspaceSize"接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用"aclnnTopk"接口执行计算。