昇腾社区首页
中文
注册

embed_query

功能描述

调用TEI服务,将用户提供的文本转换至向量。

函数原型

def embed_query(text)

输入参数说明

参数名

数据类型

是否必选

说明

text

str

必选

待向量化文本,长度范围:[1, 128 * 1024 * 1024]。

返回值说明

数据类型

说明

List[float]

text转换后的向量数组。

如果embedding模型的输出是1024维向量,最终的输出结果为(1,1024)大小的数组。