Trait nncombinator::layer::Backward

pub trait Backward<U, I, O>where
    U: UnitValue<U>,{
    fn backward(&mut self, input: I) -> O;
}

Required Methods

fn backward(&mut self, input: I) -> O

Implementors

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

impl<U, C, P, D, I, const NI: usize, const NO: usize> Backward<U, &Arr<U, NO>, Result<Arr<U, NI>, TrainingError>> 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>,
    U: Default + Clone + Copy + UnitValue<U>,
    D: Device<U> + DeviceLinear<U, C, NI, NO>,
    I: Debug + Send + Sync,