Trait einops::RearrangeFn

pub trait RearrangeFn {
    fn rearrange(&self, pattern: &str) -> Result<Self, EinopsError>
    where
        Self: Sized + Backend,
    { ... }
    fn rearrange_with_lengths(
        &self, 
        pattern: &str, 
        axes_lengths: &[(&str, usize)]
    ) -> Result<Self, EinopsError>
    where
        Self: Sized + Backend,
    { ... }
}

Trait that allows calling the rearrange operation directly on the tensor

Provided methods

fn rearrange(&self, pattern: &str) -> Result<Self, EinopsError> where
    Self: Sized + Backend, 

Apply rearrange operation using the pattern

fn rearrange_with_lengths(
    &self,
    pattern: &str,
    axes_lengths: &[(&str, usize)]
) -> Result<Self, EinopsError> where
    Self: Sized + Backend, 

Apply rearrange operation using the pattern and additional axes lengths attribute

Loading content...

Implementors

Loading content...