函数:set_nm_top_num

C函数原型

aclError aclfvSetNMTopNum(aclfvInitPara *initPara, uint32_t maxTopNumForNM)

Python函数

ret = acl.fv.set_nm_top_num(init_para, max_top_num_for_nm)

函数功能

设置N:M场景下返回的topK结果数量的最大值,如果不设置,默认topK结果数量的最大值为500。同步接口。

Atlas 200/500 A2推理产品不支持该接口。

输入说明

  • init_para:int,指定特征向量检索的初始化参数。

    需提前调用acl.fv.create_init_para接口创建aclfvInitPara类型的数据。

  • max_top_num_for_nm:int,N:M场景下返回的topK结果的最大值,取值范围:[500,4800]。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

注意事项

参考资源