昇腾社区首页
中文
注册

search

功能描述

在数据库中检索与传入的向量相似的文档。

函数原型

def search(embeddings, k, filter_dict)

参数说明

参数名

数据类型

可选/必选

说明

embeddings

List[List[float]]

必选

向量嵌入对象。向量的数量取值范围[1, 1024 * 1024)。

k

int

可选

返回的相似向量的个数,取值大于0,默认为“3”,取值范围为(0, 10000]。

filter_dict

Dict

可选

预留,MindFAISS当前暂不支持过滤检索。

返回值说明

数据类型

说明

Tuple[List[List[float]], List[List[int]]]

返回2个数据,第一个数据表示相似向量的得分列表,第二个数据表示相似向量的ID列表。