昇腾社区首页
中文
注册

trainCodeBookFromMem接口

API定义

void trainCodeBookFromMem(const AscendIndexCodeBookInitFromMemParams &codeBookInitFromMemParams) const;

功能描述

IVFSP码本训练接口。训练数据从内存中加载,如果训练速度较慢,可能是安装OpenBLAS时限制了使用单线程,可以设置环境变量export OMP_NUM_THREADS=4进行加速。

输入

const AscendIndexCodeBookInitFromMemParams &codeBookInitFromMemParams:训练码本所需的初始化参数。

输出

返回值

参数约束

了解AscendIndexCodeBookInitFromMemParams相关说明,请参见AscendIndexCodeBookInitFromMemParams