candle_einops

Trait Backend

Source 
pub trait Backend {
    type Output;

    // Required methods
    fn shape(self) -> Vec<usize>;
    fn reshape(self, shape: &[usize]) -> Self::Output;
    fn transpose(self, axes: &[usize]) -> Self::Output;
    fn reduce_axes(
        self,
        axes_operations: &mut [(usize, Operation)],
    ) -> Self::Output;
    fn add_axes(self, naxes: usize, pos2len: &[(usize, usize)]) -> Self::Output;
}

Required Associated Types§

Source

type Output

Required Methods§

Source

fn shape(self) -> Vec<usize>

Source

fn reshape(self, shape: &[usize]) -> Self::Output

Source

fn transpose(self, axes: &[usize]) -> Self::Output

Source

fn reduce_axes(self, axes_operations: &mut [(usize, Operation)]) -> Self::Output

Source

fn add_axes(self, naxes: usize, pos2len: &[(usize, usize)]) -> Self::Output

Implementors§