Trait nncombinator::layer::Loss

pub trait Loss<U>: BackwardAll<U>where
    U: UnitValue<U>,{
    fn loss<L: LossFunction<U>>(
        &mut self,
        loss: Self::LossInput,
        _lossf: &L,
        stack: Self::OutStack
    ) -> Result<(Self::OutStack, Self::LossInput), TrainingError> { ... }
}

Trait defining the calculation of the error during error back propagation.

Provided Methods

fn loss<L: LossFunction<U>>(
    &mut self,
    loss: Self::LossInput,
    _lossf: &L,
    stack: Self::OutStack
) -> Result<(Self::OutStack, Self::LossInput), TrainingError>

Error Calculation

Arguments

• loss - Lower layer error
• _lossf - loss function
• stack - Stack to store calculation results at upper layers

Errors

This function may return the following errors

• TrainingError

Implementors

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

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

impl<U, P, A, I, D, const N: usize> Loss<U> 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>,
    U: Default + Clone + Copy + UnitValue<U>,
    D: Device<U>,
    A: Activation<U, Arr<U, N>, Arr<U, N>, D>,
    I: Debug + Send + Sync,

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

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