AscendIndexVstarInitParams
API定义  | 
AscendIndexVstarInitParams(int dim, int subSpaceDim, int nlist, const std::vector<int>& deviceList, bool verbose = false, int64_t resourceSize = VSTAR_DEFAULT_MEM);  | 
|---|---|
功能描述  | 
Vstar模式初始化参数结构体。  | 
输入  | 
int dim:特征向量的维度。 int subSpaceDim:第一次降维后的维度大小。 int nlist:一级聚类的数量。 const std::vector<int>& deviceList:指定的NPU physical ID。 bool verbose:指定是否开启verbose选项,开启后部分操作提供额外的打印提示。默认值“false“。 int64_t resourceSize:设备侧预置的内存池大小,单位为Byte,计算过程中存储中间结果的内存空间,用于避免计算过程中动态申请内存造成性能波动。默认参数为头文件中定义的“VSTAR_DEFAULT_MEM”,大小为128M。该参数通过底库大小和search的batch数共同确定。  | 
输出  | 
无  | 
参数约束  | 
dim ∈ {128, 256, 512, 1024},默认值为“1024”。 subSpaceDim ∈ {32, 64, 128}。subSpaceDim必须小于dim。默认值“128“。 nlist∈ {256, 512, 1024}。默认值“1024“。 deviceList:请使用npu-smi命令查询对应的NPU卡physical ID,仅支持一个device设备ID。 resourceSize ∈ [128M, 2048M]。  |