pub trait Module: Debug + Send {
// Required method
fn forward(&self, xs: &Tensor) -> Tensor;
}
Expand description
The simplest module trait, defining a forward function.
pub trait Module: Debug + Send {
// Required method
fn forward(&self, xs: &Tensor) -> Tensor;
}
The simplest module trait, defining a forward function.