Struct tsuga::conv_layer::ConvLayer

pub struct ConvLayer {
    pub kernel: Array2<f32>,
    pub padding: usize,
    pub stride: usize,
}

A convolutional layer, which contains hyperparameters for the padding, stride, and static kernel used in the sliding-window convolution,

Fields

kernel: Array2<f32>

padding: usize

stride: usize

Implementations

impl ConvLayer

pub fn default(kernel: &Array2<f32>) -> Self

Builds the fundamental structure of a convolutional layer with a 2x2 kernel, no padding, and a stride of 1

pub fn kernel(self, kernel: &Array2<f32>) -> Self

Sets the 2D kernel used in the sliding-window convolution

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

A hyperparameter defining the stride of the sliding-window convolution

pub fn build(self) -> ConvLayer

pub fn convolve(&self, input: &Array2<f32>) -> Array2<f32>

Runs a basic sliding-window convolution operation on an NdArray Array2 structure

Trait Implementations

impl Clone for ConvLayer

fn clone(&self) -> ConvLayer

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ConvLayer

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

Formats the value using the given formatter.

impl PartialEq<ConvLayer> for ConvLayer

fn eq(&self, other: &ConvLayer) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &ConvLayer) -> bool

This method tests for !=.

impl StructuralPartialEq for ConvLayer