Struct TensorView 

pub struct TensorView<'a> {
    pub data: &'a mut [f32],
    pub shape: [usize; 5],
}

Fields

data: &'a mut [f32]

shape: [usize; 5]

Implementations

impl<'a> TensorView<'a>

pub fn is_valid_layout(&self) -> bool

pub fn len(&self) -> usize

pub fn idx_linear( &self, n: usize, c: usize, d: usize, h: usize, w: usize, ) -> Option<usize>

pub fn get( &self, n: usize, c: usize, d: usize, h: usize, w: usize, ) -> Option<f32>

pub fn get_mut( &mut self, n: usize, c: usize, d: usize, h: usize, w: usize, ) -> Option<&mut f32>

pub fn as_ptr(&self) -> *const f32

pub fn as_mut_ptr(&mut self) -> *mut f32

Trait Implementations

impl<'a> Debug for TensorView<'a>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.