Struct DenseLayer 

pub struct DenseLayer {
    pub weights: Tensor,
    pub bias: Tensor,
}

Fields

weights: Tensor

bias: Tensor

Implementations

impl DenseLayer

pub fn new(in_features: usize, out_features: usize) -> Self

pub fn forward(&self, input: &Tensor) -> Tensor

pub fn parameter_count(&self) -> usize

Trait Implementations

impl Clone for DenseLayer

fn clone(&self) -> DenseLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DenseLayer

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

Formats the value using the given formatter.