Trait Module

pub trait Module<T: Num, D: Device> {
    type Input: ModuleParameters<T, D>;
    type Output: ModuleParameters<T, D>;

    // Required method
    fn call(&self, input: Self::Input) -> Self::Output;
}

Required Associated Types

type Input: ModuleParameters<T, D>

type Output: ModuleParameters<T, D>

Required Methods

fn call(&self, input: Self::Input) -> Self::Output

Implementors

impl<T: Num, D: Device> Module<T, D> for BatchNorm2d<T, D>

type Input = Variable<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for Conv2d<T, D>

type Input = Variable<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for Dropout<T, D>

type Input = Variable<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for Linear<T, D>

type Input = Variable<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for MaxPool2d<T>

type Input = Variable<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for GRU<T, D>

type Input = GRUInput<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for LSTM<T, D>

type Input = LSTMInput<T, D>

type Output = Variable<T, D>

impl<T: Num, D: Device> Module<T, D> for RNN<T, D>

type Input = RNNLayerInput<T, D>

type Output = Variable<T, D>