下载
中文
注册

Search

API定义

APP_ERROR Search(const AscendIndexSearchParams& searchParams);

功能描述

实现AscendIndexGreat特征向量查询接口,根据输入的特征向量返回最相似的“topk”条特征的距离及ID。

输入

searchParams结构体见AscendIndexSearchParams

size_t n查询的特征向量的条数

std::vector<float>& queryData特征向量数据

int topK需要返回的最近似的结果的个数

输出

std::vector<float>& dists:查询向量与距离最近的前“topK”个向量间的距离值。

std::vector<int64_t>& labels:查询的距离最近的前“topK”个向量的ID。当有效的检索结果不足“topK”个时,剩余无效label用-1填充。

返回值

APP_ERROR:调用返回状态,具体请参见接口调用返回值参考

约束说明

  • topK ∈ (0, 4096]
  • n∈ (0, 10000]
  • queryData不能为空,且数据长度必须大于等于n * dim。
  • dists不能为空,且数据长度必须大于等于 n * topK。
  • labels不能为空,且数据长度必须大于等于 n * topK。