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