Trait rai_core::Module

pub trait Module {
    // Required methods
    fn forward(&self, input: Tensor) -> Tensor;
    fn parameters(&self) -> Vec<Tensor>;
}

Required Methods

fn forward(&self, input: Tensor) -> Tensor

fn parameters(&self) -> Vec<Tensor>

Implementors