Struct WeightMatrix 

pub struct WeightMatrix {
    pub data: Vec<f32>,
    pub shape: Vec<usize>,
    pub name: String,
}

Weight matrix with shape information

Fields

data: Vec<f32>

shape: Vec<usize>

name: String

Implementations

impl WeightMatrix

pub fn new(data: Vec<f32>, shape: Vec<usize>, name: String) -> Self

pub fn rows(&self) -> usize

pub fn cols(&self) -> usize

pub fn get_row(&self, row_idx: usize) -> &[f32]

pub fn get_row_mut(&mut self, row_idx: usize) -> &mut [f32]

Trait Implementations

impl Clone for WeightMatrix

fn clone(&self) -> WeightMatrix

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for WeightMatrix

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

Formats the value using the given formatter.