Trait Module 

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

Trait implemented by all NN modules.

Required Methods

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

Run the forward pass.

Implementors

impl Module for Dropout

impl Module for Embedding

impl Module for LayerNorm

impl Module for Linear

impl Module for RmsNorm