pub struct UnifiedTransformerBlock { /* private fields */ }Implementations§
source§impl UnifiedTransformerBlock
impl UnifiedTransformerBlock
sourcepub fn load(
vb: VarBuilder<'_>,
config: &ESMCConfig,
layer: usize,
) -> Result<Self>
pub fn load( vb: VarBuilder<'_>, config: &ESMCConfig, layer: usize, ) -> Result<Self>
Creates a new UnifiedTransformerBlock.
§Parameters
- d_model: The dimensionality of the input and output features
- n_heads: The number of attention heads
- use_geom_attn: Whether to use geometric attention
- use_plain_attn: Whether to use plain attention
- v_heads: Number of heads for geometric attention
Auto Trait Implementations§
impl Freeze for UnifiedTransformerBlock
impl !RefUnwindSafe for UnifiedTransformerBlock
impl !Send for UnifiedTransformerBlock
impl !Sync for UnifiedTransformerBlock
impl Unpin for UnifiedTransformerBlock
impl !UnwindSafe for UnifiedTransformerBlock
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
source§impl<T> IntoEither for T
impl<T> IntoEither for T
source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moresource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more