Enum DifferentiableOperation 

pub enum DifferentiableOperation {
    MatMul,
    Add,
    Mul,
    Sub,
    Div,
    Activation {
        function: ActivationFunction,
    },
    Loss {
        function: LossFunction,
    },
    Normalization {
        method: NormalizationMethod,
    },
    Convolution {
        kernel_size: usize,
        stride: usize,
        padding: usize,
    },
    Pooling {
        pool_type: PoolingType,
        kernel_size: usize,
        stride: usize,
    },
    Dropout {
        rate: f64,
    },
    Reshape {
        shape: Vec<usize>,
    },
    Concatenate {
        axis: usize,
    },
    Slice {
        start: usize,
        end: usize,
        axis: usize,
    },
    Custom {
        name: String,
        forward: fn(&[Array2<Float>]) -> Array2<Float>,
    },
}

Differentiable operations

Variants

MatMul

Matrix multiplication

Add

Element-wise addition

Mul

Element-wise multiplication

Sub

Element-wise subtraction

Div

Element-wise division

Activation

Activation functions

Fields

function: ActivationFunction

Loss

Loss functions

Fields

function: LossFunction

Normalization

Normalization

Fields

method: NormalizationMethod

Convolution

Convolution

Fields

kernel_size: usize

stride: usize

padding: usize

Pooling

Pooling

Fields

pool_type: PoolingType

kernel_size: usize

stride: usize

Dropout

Dropout

Fields

rate: f64

Reshape

Reshape

Fields

shape: Vec<usize>

Concatenate

Concatenation

Fields

axis: usize

Slice

Slice

Fields

start: usize

end: usize

axis: usize

Custom

Custom operation

Fields

name: String

forward: fn(&[Array2<Float>]) -> Array2<Float>

Trait Implementations

impl Clone for DifferentiableOperation

fn clone(&self) -> DifferentiableOperation

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DifferentiableOperation

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

Formats the value using the given formatter.