Trait neuronika::nn::DropoutInput

pub trait DropoutInput {
    type Output;
    fn dropout(self, p: f64, status: Rc<Cell<bool>>) -> Self::Output;
}

Dropout input.

This trait is implemented by Var and VarDiff.

Associated Types

type Output

Required methods

fn dropout(self, p: f64, status: Rc<Cell<bool>>) -> Self::Output

Implementors

impl<T> DropoutInput for Var<T> where
    T: Data + Forward, 

type Output = Var<DropoutNode<T>>

impl<T, U> DropoutInput for VarDiff<T, U> where
    T: Data + Forward,
    U: Gradient<Dim = T::Dim> + Overwrite + Backward, 

type Output = VarDiff<DropoutNode<T>, DropoutBackwardNode<U, T>>