昇腾社区首页
中文
注册

功能介绍

AscendIndexIVFast类,包含降维算法的三级检索AI算法,需要传入两个参数指明降维前后的维度信息,要求降维后维度能整除降维前的维度。

首先需要在GPU上训练AI聚类模型。

其次需要按照IVFSQT算子生成方式,生成三级检索所需算子。

此外,该类型也带有模糊聚类功能:入桶前,threshold参数控制底库冗余程度,搜索时建议使用batch size = 65536