AscendIndexIVFPQ
API定义 |
AscendIndexIVFPQ(int dims, int nlist, int subQuantizers, int bitsPerCode, faiss::MetricType metric, AscendIndexIVFPQConfig config = AscendIndexIVFPQConfig()) |
---|---|
功能描述 |
AscendIndexIVFPQ的构造函数,生成AscendIndexIVFPQ,此时根据“config”中配置的值设置NPU侧资源。 |
输入 |
int dims:AscendIndexIVFPQ管理的一组特征向量的维度。 int nlist:聚类中心的个数,与算子生成脚本中的“coarse_centroid_num”参数对应。 int subQuantizers:AscendIndexIVFPQ的子量化器数量。 int bitsPerCode:AscendIndexIVFPQ的bit数。 faiss::MetricType metric:AscendIndex在执行特征向量相似度检索的时候使用的距离度量类型。 AscendIndexIVFFlatConfig config:NPU侧资源配置。 |
输出 |
无 |
返回值 |
无 |
约束说明 |
|
父主题: AscendIndexIVFPQ