import { default as TeachableLLM } from '../TeachableLLM';
import { Tensor2D, Tensor3D } from '@tensorflow/tfjs-core';
export declare function meanPooling(embeddings: Tensor3D, attentionMask?: Tensor2D): Tensor2D;
export declare function sentenceEmbeddingsTensor(model: TeachableLLM, sentences: string[], batchSize?: number): Tensor2D;
export declare function sentenceEmbeddings(model: TeachableLLM, sentences: string[], batchSize?: number): Promise<number[][]>;
