Struct EmbeddingModel 

pub struct EmbeddingModel { /* private fields */ }

Implementations

impl EmbeddingModel

pub fn new( cfg: &Config, base_vb: VarBuilder<'_>, embed_vb: VarBuilder<'_>, ) -> Result<Self>

pub fn forward(&mut self, input_ids: &Tensor, mask: &Tensor) -> Result<Tensor>

pub fn forward_norm( &mut self, input_ids: &Tensor, mask: &Tensor, ) -> Result<Tensor>

Same as forward pass but normalizes the output

Trait Implementations

impl Debug for EmbeddingModel

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.