Struct syntaxdot_transformers::models::albert::AlbertEmbeddings

source · [−]

pub struct AlbertEmbeddings { /* private fields */ }

Expand description

ALBERT embeddings.

These embeddings are the same as BERT embeddings. However, we do some wrapping to ensure that the right embedding dimensionality is used.

Implementations

source

impl AlbertEmbeddings

source

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

Construct new ALBERT embeddings with the given variable store and ALBERT configuration.

source

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

Trait Implementations

source

impl Debug for AlbertEmbeddings

source

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

Formats the value using the given formatter. Read more

source

impl FallibleModuleT for AlbertEmbeddings

type Error = TransformerError

The error type.

source

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

Apply the module.

Auto Trait Implementations

impl RefUnwindSafe for AlbertEmbeddings

impl Send for AlbertEmbeddings

impl !Sync for AlbertEmbeddings

impl Unpin for AlbertEmbeddings

impl UnwindSafe for AlbertEmbeddings

Blanket Implementations

source

impl<T> Any for T where
T: 'static + ?Sized,

source

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

source

impl<T> Borrow<T> for T where
T: ?Sized,

const: unstable · source

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

source

impl<T> BorrowMut<T> for T where
T: ?Sized,

const: unstable · source

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

source

impl<T> From<T> for T

const: unstable · source

fn from(t: T) -> T

Returns the argument unchanged.

source

impl<T, U> Into for T where
U: From<T>,

const: unstable · source

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

source

impl<T> Same<T> for T

type Output = T

Should always be Self

source

impl<T, U> TryFrom for T where
U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

const: unstable · source

fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>

Performs the conversion.

source

impl<T, U> TryInto for T where
U: TryFrom<T>,

type Error = >::Error

The type returned in the event of a conversion error.

const: unstable · source

fn try_into(self) -> Result<U, >::Error>

Performs the conversion.

Struct syntaxdot_transformers::models::albert::AlbertEmbeddings

Implementations

impl AlbertEmbeddings

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

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 AlbertEmbeddings

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

impl FallibleModuleT for AlbertEmbeddings

type Error = TransformerError

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

Auto Trait Implementations

impl RefUnwindSafe for AlbertEmbeddings

impl Send for AlbertEmbeddings

impl !Sync for AlbertEmbeddings

impl Unpin for AlbertEmbeddings

impl UnwindSafe for AlbertEmbeddings

Blanket Implementations

impl<T> Any for T where
T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for T where
T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for T where
T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for T where
U: From<T>,

fn into(self) -> U

impl<T> Same<T> for T

type Output = T

impl<T, U> TryFrom<U> for T where
U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for T where
U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for T where
V: MultiLane<T>,

fn vzip(self) -> V

Struct syntaxdot_transformers::models::albert::AlbertEmbeddings

Implementations

impl AlbertEmbeddings

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

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 AlbertEmbeddings

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

impl FallibleModuleT for AlbertEmbeddings

type Error = TransformerError

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

Auto Trait Implementations

impl RefUnwindSafe for AlbertEmbeddings

impl Send for AlbertEmbeddings

impl !Sync for AlbertEmbeddings

impl Unpin for AlbertEmbeddings

impl UnwindSafe for AlbertEmbeddings

Blanket Implementations

impl<T> Any for T where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for T where T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for T where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for T where U: From<T>,

fn into(self) -> U

impl<T> Same<T> for T

type Output = T

impl<T, U> TryFrom<U> for T where U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for T where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for T where V: MultiLane<T>,

fn vzip(self) -> V

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

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

impl<T> Any for T where
T: 'static + ?Sized,

impl<T> Borrow<T> for T where
T: ?Sized,

impl<T> BorrowMut<T> for T where
T: ?Sized,

impl<T, U> Into<U> for T where
U: From<T>,

impl<T, U> TryFrom<U> for T where
U: Into<T>,

impl<T, U> TryInto<U> for T where
U: TryFrom<T>,

impl<V, T> VZip<V> for T where
V: MultiLane<T>,