昇腾社区首页
中文
注册

AscendIndexIVFSP

API定义

AscendIndexIVFSP(int dims, int nonzeroNum, int nlist, faiss::ScalarQuantizer::QuantizerType qType = ScalarQuantizer::QuantizerType::QT_8bit, faiss::MetricType metric = MetricType::METRIC_L2, AscendIndexIVFSPConfig config = AscendIndexIVFSPConfig());

功能描述

AscendIndexIVFSP的构造函数,根据“config”中配置的值设置Device侧资源。

输入

  • int dims:AscendIndexIVFSP管理的一组特征向量的维度。
  • int nonzeroNum:特征向量压缩降维后非零维度个数。
  • int nlist:聚类中心的个数,与IVFSP章节的“IVFSP业务算子模型文件生成”中的<centroid num>参数值对应。
  • faiss::ScalarQuantizer::QuantizerType qType:标量量化类型,当前仅支持“ScalarQuantizer::QuantizerType::QT_8bit”。
  • faiss::MetricType metric:AscendIndex在执行特征向量相似度检索时使用的距离度量类型。当前“faiss::MetricType metric”仅支持“METRIC_L2”。
  • AscendIndexIVFSPConfig:Device侧资源配置。

输出

返回值

约束说明

  • 当dims ∈ {64, 128, 256}时,nlist∈ {256, 512, 1024, 2048, 4096, 8192, 16384};当dims ∈ {512, 768}时,nlist∈ {256, 512, 1024, 2048}。
  • “nonzeroNum”需为16的倍数且小于等于min(128, dims)。
  • metric ∈ {faiss::MetricType::METRIC_L2}。