Trait nncombinator::layer::ForwardAll

pub trait ForwardAll {
    type Input: Debug;
    type Output: Debug;

    fn forward_all(
        &self,
        input: Self::Input
    ) -> Result<Self::Output, EvaluateError>;
}

Trait defining the implementation of forward propagation of neural networks

Required Associated Types

type Input: Debug

Input to this layer of the neural network

type Output: Debug

Output from this layer of the neural network

Required Methods

fn forward_all(&self, input: Self::Input) -> Result<Self::Output, EvaluateError>

Forward propagation

Arguments

• input - input

Errors

This function may return the following errors

• EvaluateError

Implementors

impl<U, C, P, D, I, const NI: usize, const NO: usize> ForwardAll for DiffLinearLayer<U, C, P, D, I, NI, NO>where
    P: ForwardAll<Input = I, Output = DiffInput<DiffArr<U, NI>, U, NI, NO>> + BackwardAll<U, LossInput = Arr<U, NI>> + PreTrain<U> + Loss<U>,
    U: Default + Clone + Copy + UnitValue<U>,
    C: Index<(usize, usize), Output = U>,
    D: Device<U> + DeviceLinear<U, C, NI, NO>,
    I: Debug + Send + Sync,

type Input = I

type Output = Arr<U, NO>

impl<U, C, P, D, I, const NI: usize, const NO: usize> ForwardAll for LinearLayer<U, C, P, D, I, NI, NO>where
    P: ForwardAll<Input = I, Output = Arr<U, NI>> + BackwardAll<U, LossInput = Arr<U, NI>> + PreTrain<U> + Loss<U>,
    D: Device<U> + DeviceLinear<U, C, NI, NO>,
    U: Default + Clone + Copy + Send + UnitValue<U>,
    I: Debug + Send + Sync,

type Input = I

type Output = Arr<U, NO>

impl<U, O, LI> ForwardAll for InputLayer<U, O, LI>where
    U: UnitValue<U>,
    O: Debug + Send + Sync + 'static,
    LI: Debug,

type Input = O

type Output = O

impl<U, P, A, I, PI, D> ForwardAll for ActivationLayer<U, P, A, I, PI, D>where
    P: ForwardAll<Input = I, Output = PI> + BackwardAll<U, LossInput = PI> + PreTrain<U> + Loss<U>,
    U: Default + Clone + Copy + UnitValue<U>,
    D: Device<U>,
    A: Activation<U, PI, PI, D>,
    PI: Debug,
    I: Debug + Send + Sync,

type Input = I

type Output = PI

impl<U, P, D, I, IO> ForwardAll for LinearOutputLayer<U, P, D, I, IO>where
    P: ForwardAll<Input = I, Output = IO> + BackwardAll<U, LossInput = IO> + PreTrain<U> + Loss<U>,
    U: Default + Clone + Copy + UnitValue<U>,
    D: Device<U>,
    IO: Debug,
    I: Debug + Send + Sync,

type Input = I

type Output = IO