Skip to main content

LayerCpu

Type Alias LayerCpu 

Source
pub type LayerCpu = Layer<CpuLinAlg>;
Expand description

Type alias: CPU-backed layer.

Aliased Type§

pub struct LayerCpu {
    pub weights: Matrix,
    pub bias: Vec<f64>,
    pub activation: Activation,
}

Fields§

§weights: Matrix

Weight matrix of shape [output_size × input_size].

§bias: Vec<f64>

Bias vector of length output_size.

§activation: Activation

Activation function applied element-wise after the linear transform.