Struct Dense 

pub struct Dense { /* private fields */ }

Dense (fully connected) layer

Implementations

impl Dense

pub fn new(units: usize) -> Self

Create new dense layer

pub fn activation(self, activation: ActivationFunction) -> Self

Set activation function

pub fn use_bias(self, use_bias: bool) -> Self

Set use bias

pub fn name(self, name: impl Into<String>) -> Self

Set layer name

pub fn kernel_initializer(self, initializer: InitializerType) -> Self

Set kernel initializer

Trait Implementations

impl KerasLayer for Dense

fn build(&mut self, input_shape: &[usize]) -> Result<()>

Build the layer (called during model compilation)

fn call(&self, inputs: &ArrayD<f64>) -> Result<ArrayD<f64>>

Forward pass through the layer

fn compute_output_shape(&self, input_shape: &[usize]) -> Vec<usize>

Compute output shape given input shape

fn name(&self) -> &str

Get layer name

fn get_weights(&self) -> Vec<ArrayD<f64>>

Get trainable parameters

fn set_weights(&mut self, weights: Vec<ArrayD<f64>>) -> Result<()>

Set trainable parameters

fn built(&self) -> bool

Check if layer is built

fn count_params(&self) -> usize

Get number of parameters