Trait ApplyGradient

pub trait ApplyGradient<Delta, T> {
    type Output;

    // Required methods
    fn apply_gradient(&mut self, grad: &Delta, lr: T) -> Result<Self::Output>;
    fn apply_gradient_with_decay(
        &mut self,
        grad: &Delta,
        lr: T,
        decay: T,
    ) -> Result<Self::Output>;
}

A trait declaring basic gradient-related routines for a neural network

Required Associated Types

type Output

Required Methods

fn apply_gradient(&mut self, grad: &Delta, lr: T) -> Result<Self::Output>

fn apply_gradient_with_decay( &mut self, grad: &Delta, lr: T, decay: T, ) -> Result<Self::Output>

Implementations on Foreign Types

impl<A, S, T, D> ApplyGradient<ArrayBase<T, D>, A> for ArrayBase<S, D>

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

type Output = ()

fn apply_gradient( &mut self, grad: &ArrayBase<T, D>, lr: A, ) -> Result<Self::Output>

fn apply_gradient_with_decay( &mut self, grad: &ArrayBase<T, D>, lr: A, decay: A, ) -> Result<Self::Output>

Implementors

impl<A, S, T, D> ApplyGradient<ParamsBase<T, D>, A> for ParamsBase<S, D>

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

type Output = ()