rustyml 0.13.0

A high-performance machine learning & deep learning library in pure Rust, offering ML algorithms and neural network support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

//! Convolutional layers and shared padding types
//!
//! Re-exports the 1D/2D/3D, depthwise, and separable convolution layers, and
//! defines the [`PaddingType`] used to control spatial padding

/// Padding method used by convolutional and pooling layers
///
/// Determines how the input is padded before the operation is applied. Defaults to
/// [`PaddingType::Valid`] (no padding)
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum PaddingType {
    /// No padding is applied; the convolution is computed only where the filter
    /// fully overlaps the input, producing an output with reduced dimensions
    #[default]
    Valid,

    /// Zeros are added around the input borders so the output keeps the same
    /// spatial dimensions as the input (when stride is 1)
    Same,
}

/// 1D convolutional layer
pub mod conv_1d;
/// 2D convolutional layer
pub mod conv_2d;
/// 3D convolutional layer
pub mod conv_3d;
/// Dimension-generic convolution engine shared by Conv1D, Conv2D, and Conv3D
pub(crate) mod convolution_engine;
/// 2D depthwise convolutional layer
pub mod depthwise_conv_2d;
/// 2D separable convolutional layer
pub mod separable_conv_2d;
/// Input validation helpers for convolutional layers
mod validation;

pub use conv_1d::Conv1D;
pub use conv_2d::Conv2D;
pub use conv_3d::Conv3D;
pub use depthwise_conv_2d::DepthwiseConv2D;
pub use separable_conv_2d::SeparableConv2D;