Struct Conv2D 

pub struct Conv2D { /* private fields */ }

Convolutional layer.

Implementations

impl Conv2D

pub fn new( name: &str, filters: Box<dyn ArrayProtocol>, bias: Option<Box<dyn ArrayProtocol>>, stride: (usize, usize), padding: (usize, usize), activation: Option<ActivationFunc>, ) -> Self

Create a new convolutional layer.

pub fn withshape( name: &str, filter_height: usize, filter_width: usize, in_channels: usize, out_channels: usize, stride: (usize, usize), padding: (usize, usize), withbias: bool, activation: Option<ActivationFunc>, ) -> Self

Create a new convolutional layer with randomly initialized weights.

Trait Implementations

impl Layer for Conv2D

fn layer_type(&self) -> &str

Forward pass through the layer. Get the layer type name for serialization.

fn forward( &self, inputs: &dyn ArrayProtocol, ) -> Result<Box<dyn ArrayProtocol>, OperationError>

fn parameters(&self) -> Vec<Box<dyn ArrayProtocol>>

Get the layer’s parameters.

fn parameters_mut(&mut self) -> Vec<&mut Box<dyn ArrayProtocol>>

Get mutable references to the layer’s parameters.

fn update_parameter( &mut self, name: &str, value: Box<dyn ArrayProtocol>, ) -> Result<(), OperationError>

Update a specific parameter by name

fn parameter_names(&self) -> Vec<String>

Get parameter names

fn train(&mut self)

Set the layer to training mode.

fn eval(&mut self)

Set the layer to evaluation mode.

fn is_training(&self) -> bool

Check if the layer is in training mode.

fn name(&self) -> &str

Get the layer’s name.