search_batched_parallel
接口定义
def search_batched_parallel(queries: numpy.ndarray, final_num_neighbors: int, pre_reorder_num_neighbors:int, leaves_to_search: int, batch_size: int) -> tuple(index, distance)
接口用途
接口并行批量query搜索接口，多线程并发执行（与开源算法接口保持一致）。
参数说明
参数名称 |
数据类型 |
描述 |
取值范围 |
|---|---|---|---|
queries |
numpy.ndarray |
批次内的Queries。 |
限制非空。 |
final_num_neighbors |
int |
最终返回结果数量，TopK。 |
大于等于1。 |
pre_reorder_num_neighbors |
int |
重排前保存结果数量。 |
大于等于final_num_neighbors，其中“final_num_neighbors”为最终返回结果数量。 |
leaves_to_search |
int |
搜索子空间数量。 |
[1, num_leaves]，其中“num_leaves”为IVF倒排索引分区总子空间个数。 |
batch_size |
int |
并行时自动分批时优先批次大小，默认值256。 |
大于等于1。 |
返回值
数据类型 |
说明 |
|---|---|
tuple(index, distance) |
index为包含final_num_neighbors个底库ID的numpy.ndarray；distance为包含final_num_neighbors个距离的numpy.ndarray。 |
父主题： Python接口