torch_frame.config.TextEmbedderConfig

class TextEmbedderConfig(text_embedder: Callable[[list[str]], Tensor], batch_size: int | None = None)

基础类：object

文本嵌入模型，将字符串/句子列表映射为PyTorch张量嵌入。

Parameters:

• text_embedder (callable) – 一个可调用的文本嵌入器，它接受一个字符串列表作为输入，并输出该字符串列表的PyTorch张量嵌入。

• batch_size (int, optional) – 编码句子时使用的批量大小。如果设置为 None，文本嵌入将以全批量的方式获取。（默认值：None）