search
接口定义
def search(nq: int, xq: numpy.ndarray, topk: int, distances: numpy.ndarray, labels: numpy.ndarray, num_threads: int) -> int
接口用途
图检索。
参数说明
参数名称 |
数据类型 |
描述 |
取值范围 |
|---|---|---|---|
nq |
int |
查询向量集数据量。 |
大于等于1。建议小于10亿。 |
xq |
numpy.ndarray |
查询向量集数据。 |
限制非空,期待长度为nq * dim(数据量 * 特征维度)。 |
topk |
int |
最近邻的数量。 |
[1,nb],其中“nb”为向量底库数据量大小。 |
distances |
numpy.ndarray |
距离数组。 |
限制非空,期待长度为nq * topk。 |
labels |
numpy.ndarray |
检索结果的ID数组。 |
限制非空,期待长度为nq * topk。 |
num_threads |
int |
检索时使用的线程数。 |
[1,std::thread::hardware_concurrency()],其中std::thread::hardware_concurrency()为当前系统支持的并发线程数。 |
返回值
数据类型 |
说明 |
|---|---|
int |
接口由底层C++接口封装而来,异常抛出由底层C++接口控制。 |
父主题: Python