昇腾社区首页
中文
注册

函数:set_nm_top_num

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

x

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

x

Atlas 训练系列产品

x

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

x

Atlas 200/300/500 推理产品

x

功能说明

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

Atlas 推理系列产品 中的 Atlas 推理系列加速模块产品 不支持该接口。

函数原型

  • C函数原型
    1
    aclError aclfvSetNMTopNum(aclfvInitPara *initPara, uint32_t maxTopNumForNM)
    
  • python函数
    1
    ret = acl.fv.set_nm_top_num(init_para, max_top_num_for_nm)
    

参数说明

参数名

说明

init_para

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

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

max_top_num_for_nm

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

返回值说明

返回值

说明

ret

int,错误码。

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

资源参考

接口调用流程及示例代码,参见特征向量检索