Trait rust_bert::bert::BertEmbedding

pub trait BertEmbedding {
    fn new<'p, P>(p: P, config: &BertConfig) -> Self
    where
        P: Borrow<Path<'p>>;
    fn forward_t(
        &self,
        input_ids: Option<&Tensor>,
        token_type_ids: Option<&Tensor>,
        position_ids: Option<&Tensor>,
        input_embeds: Option<&Tensor>,
        train: bool
    ) -> Result<Tensor, RustBertError>;
}

BertEmbedding trait (for use in BertModel or RoBERTaModel)

Defines an interface for the embedding layers in BERT-based models

Required Methods

fn new<'p, P>(p: P, config: &BertConfig) -> Selfwhere
    P: Borrow<Path<'p>>,

fn forward_t(
    &self,
    input_ids: Option<&Tensor>,
    token_type_ids: Option<&Tensor>,
    position_ids: Option<&Tensor>,
    input_embeds: Option<&Tensor>,
    train: bool
) -> Result<Tensor, RustBertError>

Implementors

impl BertEmbedding for RobertaEmbeddings

impl BertEmbedding for BertEmbeddings