Type Alias sbert::SBertHF

pub type SBertHF = SBert<HFTokenizer>;

Aliased Type

struct SBertHF { /* private fields */ }

Implementations

impl<T> SBert<T>where T: Tokenizer + Send + Sync,

pub fn new<P>(root: P) -> Result<Self, Error>where P: Into<PathBuf>,

pub fn forward<S, B>( &self, input: &[S], batch_size: B ) -> Result<Vec<Embeddings>, Error>where S: AsRef<str>, B: Into<Option<usize>>,

pub fn forward_with_attention<S, B>( &self, input: &[S], batch_size: B ) -> Result<(Vec<Embeddings>, Attentions), Error>where S: AsRef<str>, B: Into<Option<usize>>,

pub fn tokenizer(&self) -> Arc<T>