Struct syntaxdot_transformers::models::albert::AlbertEncoder

source · [−]

pub struct AlbertEncoder { /* private fields */ }

Expand description

ALBERT encoder.

This encoder uses the BERT encoder with two modifications:

The embeddings are projected to fit the hidden layer size.
Weights are shared between layers.

Implementations

source

impl AlbertEncoder

source

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

Trait Implementations

source

impl Debug for AlbertEncoder

source

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

Formats the value using the given formatter. Read more

source

impl Encoder for AlbertEncoder

source

fn encode(
 &self,
 input: &Tensor,
 attention_mask: Option<&Tensor>,
 train: bool
) -> Result<Vec<LayerOutput>, TransformerError>

Apply the encoder. Read more

source

fn n_layers(&self) -> i64

Get the number of layers that is returned by the encoder.

Auto Trait Implementations

impl RefUnwindSafe for AlbertEncoder

impl Send for AlbertEncoder

impl !Sync for AlbertEncoder

impl Unpin for AlbertEncoder

impl UnwindSafe for AlbertEncoder

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::AlbertEncoder

Implementations

impl AlbertEncoder

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

Trait Implementations

impl Debug for AlbertEncoder

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

impl Encoder for AlbertEncoder

fn encode(
    &self,
    input: &Tensor,
    attention_mask: Option<&Tensor>,
    train: bool
) -> Result<Vec<LayerOutput>, TransformerError>

fn n_layers(&self) -> i64

Auto Trait Implementations

impl RefUnwindSafe for AlbertEncoder

impl Send for AlbertEncoder

impl !Sync for AlbertEncoder

impl Unpin for AlbertEncoder

impl UnwindSafe for AlbertEncoder

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::AlbertEncoder

Implementations

impl AlbertEncoder

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

Trait Implementations

impl Debug for AlbertEncoder

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

impl Encoder for AlbertEncoder

fn encode( &self, input: &Tensor, attention_mask: Option<&Tensor>, train: bool) -> Result<Vec<LayerOutput>, TransformerError>

fn n_layers(&self) -> i64

Auto Trait Implementations

impl RefUnwindSafe for AlbertEncoder

impl Send for AlbertEncoder

impl !Sync for AlbertEncoder

impl Unpin for AlbertEncoder

impl UnwindSafe for AlbertEncoder

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>

fn encode(
&self,
input: &Tensor,
attention_mask: Option<&Tensor>,
train: bool
) -> Result<Vec<LayerOutput>, 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>,