Struct RobertaEmbeddings 

pub struct RobertaEmbeddings { /* private fields */ }

RoBERTa and XLM-RoBERTa embeddings.

Implementations

impl RobertaEmbeddings

pub fn new<'a>( vs: impl Borrow<PathExt<'a>>, config: &BertConfig, ) -> Result<RobertaEmbeddings, TransformerError>

Construct new RoBERTa embeddings with the given variable store and Bert configuration.

pub fn forward( &self, input_ids: &Tensor, token_type_ids: Option<&Tensor>, position_ids: Option<&Tensor>, train: bool, ) -> Result<Tensor, TransformerError>

Trait Implementations

impl Debug for RobertaEmbeddings

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

Formats the value using the given formatter.

impl FallibleModuleT for RobertaEmbeddings

type Error = TransformerError

The error type.

fn forward_t(&self, input: &Tensor, train: bool) -> Result<Tensor, Self::Error>

Apply the module.