Trait candle_nn::ModuleT

pub trait ModuleT {
    // Required method
    fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor, Error>;
}

Required Methods

fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor, Error>

Implementors

impl ModuleT for BatchNorm

impl ModuleT for Dropout

impl<'a> ModuleT for FuncT<'a>

impl<M> ModuleT for M

where M: Module,