Struct NeuralParameters

pub struct NeuralParameters<F: Float + ScalarOperand> {
    pub parameters: Vec<Array1<F>>,
    pub gradients: Vec<Array1<F>>,
    pub names: Vec<String>,
}

Neural network parameter container

Fields

parameters: Vec<Array1<F>>

Parameter vectors for each layer

gradients: Vec<Array1<F>>

Gradient vectors for each layer

names: Vec<String>

Parameter names/ids for tracking

Implementations

impl<F: Float + ScalarOperand> NeuralParameters<F>

pub fn new() -> Self

Create new neural parameters container

pub fn add_parameter(&mut self, name: String, param: Array1<F>)

Add a parameter vector

pub fn total_parameters(&self) -> usize

Get total number of parameters

pub fn flatten_parameters(&self) -> Array1<F>

Flatten all parameters into a single vector

pub fn flatten_gradients(&self) -> Array1<F>

Flatten all gradients into a single vector

pub fn update_from_flat(&mut self, flat_params: &Array1<F>)

Update parameters from flattened vector

pub fn update_gradients_from_flat(&mut self, flat_grads: &Array1<F>)

Update gradients from flattened vector

Trait Implementations

impl<F: Clone + Float + ScalarOperand> Clone for NeuralParameters<F>

fn clone(&self) -> NeuralParameters<F>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<F: Debug + Float + ScalarOperand> Debug for NeuralParameters<F>

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

Formats the value using the given formatter.

impl<F: Float + ScalarOperand> Default for NeuralParameters<F>

fn default() -> Self

Returns the “default value” for a type.