Trait Backward

pub trait Backward<X, Y> {
    type HParam;
    type Output;

    // Required method
    fn backward(
        &mut self,
        input: &X,
        delta: &Y,
        gamma: Self::HParam,
    ) -> Result<Self::Output>;
}

A simple trait denoting a single backward pass through a layer of a neural network; the trait

Required Associated Types

type HParam

type Output

Required Methods

fn backward( &mut self, input: &X, delta: &Y, gamma: Self::HParam, ) -> Result<Self::Output>

Implementors

impl<A, S, T> Backward<ArrayBase<S, Dim<[usize; 1]>>, ArrayBase<T, Dim<[usize; 0]>>> for Params<A, Ix1>

where A: Float + FromPrimitive + ScalarOperand, S: Data<Elem = A>, T: Data<Elem = A>,

type HParam = A

type Output = A

impl<A, S, T> Backward<ArrayBase<S, Dim<[usize; 1]>>, ArrayBase<T, Dim<[usize; 1]>>> for Params<A, Ix2>

where A: Float + FromPrimitive + ScalarOperand, S: Data<Elem = A>, T: Data<Elem = A>,

type HParam = A

type Output = A

impl<A, S, T> Backward<ArrayBase<S, Dim<[usize; 2]>>, ArrayBase<T, Dim<[usize; 1]>>> for Params<A, Ix1>

where A: Float + FromPrimitive + ScalarOperand, S: Data<Elem = A>, T: Data<Elem = A>,

type HParam = A

type Output = A

impl<A, S, T> Backward<ArrayBase<S, Dim<[usize; 2]>>, ArrayBase<T, Dim<[usize; 2]>>> for Params<A, Ix2>

where A: Float + FromPrimitive + ScalarOperand, S: Data<Elem = A>, T: Data<Elem = A>,

type HParam = A

type Output = A