Struct Conv2D 

pub struct Conv2D { /* private fields */ }

Conv2D layer (Keras-compatible)

Implementations

impl Conv2D

pub fn new(filters: usize, kernel_size: (usize, usize)) -> Self

Create new Conv2D layer

pub fn strides(self, strides: (usize, usize)) -> Self

Set strides

pub fn padding(self, padding: &str) -> Self

Set padding

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

Set activation

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

Set use bias

pub fn name(self, name: &str) -> Self

Set layer name

Trait Implementations

impl KerasLayer for Conv2D

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

Forward pass through the layer

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

Build the layer (called during model compilation)

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

Compute output shape given input shape

fn count_params(&self) -> usize

Get number of parameters

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 name(&self) -> &str

Get layer name