Trait nncombinator::layer::BatchForwardBase

pub trait BatchForwardBase: ForwardAll {
    type BatchInput: Debug;
    type BatchOutput: Debug;
}

Required Associated Types

type BatchInput: Debug

type BatchOutput: Debug

Implementors

impl<U, C, P, D, I, const NI: usize, const NO: usize> BatchForwardBase 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> + BatchForwardBase<BatchInput = VecArr<U, I>, BatchOutput = VecArr<U, Arr<U, NI>>>,
    U: Default + Clone + Copy + Send + UnitValue<U>,
    D: Device<U>,
    I: Debug + Send + Sync,
    Self: ForwardAll,

type BatchInput = VecArr<U, I>

type BatchOutput = VecArr<U, Arr<U, NO>>

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

type BatchInput = VecArr<U, O>

type BatchOutput = VecArr<U, O>

impl<U, P, A, I, D, const N: usize> BatchForwardBase for ActivationLayer<U, P, A, I, Arr<U, N>, D>where
    P: PreTrain<U> + ForwardAll<Input = I, Output = Arr<U, N>> + BackwardAll<U, LossInput = Arr<U, N>> + Loss<U> + BatchForwardBase<BatchInput = VecArr<U, I>, BatchOutput = VecArr<U, Arr<U, N>>> + BatchPreTrainBase<U> + BatchBackward<U> + BatchLoss<U, BatchLossInput = VecArr<U, Arr<U, N>>>,
    U: Default + Clone + Copy + UnitValue<U>,
    D: Device<U>,
    A: Activation<U, Arr<U, N>, Arr<U, N>, D>,
    I: Debug + Send + Sync,

type BatchInput = VecArr<U, I>

type BatchOutput = VecArr<U, Arr<U, N>>

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

type BatchInput = VecArr<U, I>

type BatchOutput = VecArr<U, IO>