下载
中文
注册

AscendIndexIVFSQ

API定义

AscendIndexIVFSQ(int dims, int nlist, faiss::MetricType metric, AscendIndexIVFSQConfig config);

功能描述

AscendIndexIVFSQ的构造函数,生成AscendIndexIVFSQ,此时根据“config”中配置的值设置Device侧资源。此接口不执行初始化,由子类执行初始化相关功能,后续会废弃此接口,请勿使用。

输入

int dims:AscendIndexIVFSQ管理的一组特征向量的维度。

int nlist:聚类中心的个数,与算子生成脚本中的“coarse_centroid_num”参数对应。

faiss::MetricType metric:AscendIndex在执行特征向量相似度检索的时候使用的距离度量类型。

AscendIndexIVFSQConfig config:Device侧资源配置。

输出

返回值

约束说明

  • dims ∈ {64, 128, 256, 384, 512}
  • nlist ∈ {1024, 2048, 4096, 8192, 16384, 32768}
  • metric ∈ {faiss::MetricType::METRIC_L2,faiss::MetricType::METRIC_INNER_PRODUCT}