pub struct SinusoidalPositionalEncoding<F: Float + Debug + NumAssign> { /* private fields */ }Expand description
Sinusoidal Positional Encoding
Implements the classic positional encoding from “Attention Is All You Need”: PE(pos, 2i) = sin(pos / 10000^(2i/d_model)) PE(pos, 2i+1) = cos(pos / 10000^(2i/d_model))
This encoding has several desirable properties:
- Deterministic and parameter-free
- Each position gets a unique encoding
- Can extrapolate to longer sequences than seen during training
- Allows the model to attend to relative positions
Implementations§
Source§impl<F: Float + Debug + NumAssign> SinusoidalPositionalEncoding<F>
impl<F: Float + Debug + NumAssign> SinusoidalPositionalEncoding<F>
Sourcepub fn new(d_model: usize, max_len: usize) -> Self
pub fn new(d_model: usize, max_len: usize) -> Self
Create a new sinusoidal positional encoding
§Arguments
d_model- Model dimension (must be even)max_len- Maximum sequence length
Sourcepub fn with_dropout(d_model: usize, max_len: usize, dropout: F) -> Self
pub fn with_dropout(d_model: usize, max_len: usize, dropout: F) -> Self
Create with dropout
Sourcepub fn params(&self) -> Vec<&Array<F, IxDyn>> ⓘ
pub fn params(&self) -> Vec<&Array<F, IxDyn>> ⓘ
Get parameters (sinusoidal encoding has no trainable parameters)
Sourcepub fn set_training(&mut self, _training: bool)
pub fn set_training(&mut self, _training: bool)
Set training mode (no-op for sinusoidal encoding as it has no trainable parameters)
Trait Implementations§
Source§impl<F: Clone + Float + Debug + NumAssign> Clone for SinusoidalPositionalEncoding<F>
impl<F: Clone + Float + Debug + NumAssign> Clone for SinusoidalPositionalEncoding<F>
Source§fn clone(&self) -> SinusoidalPositionalEncoding<F>
fn clone(&self) -> SinusoidalPositionalEncoding<F>
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl<F: Float + Debug + NumAssign> PositionalEncoding<F> for SinusoidalPositionalEncoding<F>
impl<F: Float + Debug + NumAssign> PositionalEncoding<F> for SinusoidalPositionalEncoding<F>
Source§fn encode(&self, seq_len: usize) -> Array2<F>
fn encode(&self, seq_len: usize) -> Array2<F>
Encode positions for a sequence of given length Read more
Source§fn apply(&self, input: &Array3<F>) -> Result<Array3<F>>
fn apply(&self, input: &Array3<F>) -> Result<Array3<F>>
Apply positional encoding to an input tensor Read more
Source§fn clone_box(&self) -> Box<dyn PositionalEncoding<F> + Send + Sync>
fn clone_box(&self) -> Box<dyn PositionalEncoding<F> + Send + Sync>
Clone the positional encoding into a boxed trait object
Auto Trait Implementations§
impl<F> Freeze for SinusoidalPositionalEncoding<F>where
F: Freeze,
impl<F> RefUnwindSafe for SinusoidalPositionalEncoding<F>where
F: RefUnwindSafe,
impl<F> Send for SinusoidalPositionalEncoding<F>where
F: Send,
impl<F> Sync for SinusoidalPositionalEncoding<F>where
F: Sync,
impl<F> Unpin for SinusoidalPositionalEncoding<F>where
F: Unpin,
impl<F> UnsafeUnpin for SinusoidalPositionalEncoding<F>where
F: UnsafeUnpin,
impl<F> UnwindSafe for SinusoidalPositionalEncoding<F>where
F: UnwindSafe + RefUnwindSafe,
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.