pub struct SinusoidalPositionEncoding {
pub config: PositionEncodingConfig,
}Expand description
Sinusoidal position encoding
Fields§
§config: PositionEncodingConfigConfiguration
Implementations§
Source§impl SinusoidalPositionEncoding
impl SinusoidalPositionEncoding
Sourcepub fn new(config: PositionEncodingConfig) -> Result<Self>
pub fn new(config: PositionEncodingConfig) -> Result<Self>
Create a new sinusoidal position encoding
Sourcepub fn build_encoding_graph(
&self,
graph: &mut EinsumGraph,
) -> Result<Vec<usize>>
pub fn build_encoding_graph( &self, graph: &mut EinsumGraph, ) -> Result<Vec<usize>>
Build einsum graph for sinusoidal position encoding
Input tensors:
- 0: x (input) [batch, seq_len, d_model]
- 1: position_ids [batch, seq_len] (optional, will use 0..seq_len if not provided)
Output tensors:
- output: [batch, seq_len, d_model] (x + position_encoding)
Trait Implementations§
Source§impl Clone for SinusoidalPositionEncoding
impl Clone for SinusoidalPositionEncoding
Source§fn clone(&self) -> SinusoidalPositionEncoding
fn clone(&self) -> SinusoidalPositionEncoding
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 moreAuto Trait Implementations§
impl Freeze for SinusoidalPositionEncoding
impl RefUnwindSafe for SinusoidalPositionEncoding
impl Send for SinusoidalPositionEncoding
impl Sync for SinusoidalPositionEncoding
impl Unpin for SinusoidalPositionEncoding
impl UnwindSafe for SinusoidalPositionEncoding
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