Trait Module
tch::nn
pub trait Module { fn forward(&self, xs: &Tensor) -> Tensor; }
The simplest module trait, defining a forward function.
fn forward(&self, xs: &Tensor) -> Tensor
impl Module for Conv2D
impl Module for Linear
impl Module for Sequential
impl<'a> Module for Func<'a>