Struct Module 

pub struct Module {
    pub parameters: Arc<Mutex<Vec<ShareTensor>>>,
    pub initialization: WeightInitialization,
    pub eval_handlers: Vec<Arc<Mutex<bool>>>,
    pub rng: ChaCha8Rng,
}

Fields

parameters: Arc<Mutex<Vec<ShareTensor>>>

initialization: WeightInitialization

eval_handlers: Vec<Arc<Mutex<bool>>>

rng: ChaCha8Rng

Implementations

impl Module

pub fn init() -> Module

pub fn parameters(&self) -> Arc<Mutex<Vec<ShareTensor>>>

pub fn train(&self)

pub fn eval(&self)

pub fn update_seed(&mut self, seed: u64)

pub fn update_initialization(&mut self, initialization: WeightInitialization)

impl Module

pub fn liniar_init(&mut self, input: usize, output: usize) -> Linear

impl Module

pub fn dropout_init(&mut self, p: f64) -> Dropout

impl Module

pub fn batch_norm_init( &mut self, channel_features: usize, input_dimension: usize, ) -> BatchNorm

impl Module

pub fn layer_norm_init(&mut self, layer_shape: &[usize]) -> LayerNorm

impl Module

pub fn embedding_init(&mut self, vocab_num: usize, hidden: usize) -> Embedding

impl Module

pub fn lstm_init(&mut self, hidden: usize) -> Lstm

impl Module

pub fn gru_init(&mut self, hidden: usize) -> Gru

impl Module

pub fn add_parameter(&self, tensor: &Tensor)

pub fn init_rand_parameter(&mut self, shape: Vec<usize>) -> Tensor

impl Module

pub fn random_initialization(&mut self) -> f32

impl Module

pub fn glorot_initialization(&mut self, input: usize, output: usize) -> f32

impl Module

pub fn he_initialization(&mut self, input: usize) -> f32

impl Module

pub fn save(&self, path: &str)

pub fn load_save(&self, path: &str)