Struct ModuleBase 

pub struct ModuleBase {
    pub parameters: HashMap<String, Parameter>,
    /* private fields */
}

Base module implementation helper

Fields

parameters: HashMap<String, Parameter>

Implementations

impl ModuleBase

pub fn new() -> Self

pub fn training(&self) -> bool

Check if in training mode

pub fn set_training(&mut self, training: bool)

Set training mode

pub fn apply_to_parameters<F>(&mut self, f: F) -> Result<()>

where F: Fn(&mut Parameter) -> Result<()>,

Apply function to all parameters in this module

pub fn apply_to_modules<F>(&mut self, f: F) -> Result<()>

where F: Fn(&mut dyn Module) -> Result<()>,

Apply function to all modules recursively

pub fn children(&self) -> Vec<&dyn Module>

Get children modules as references

pub fn named_children(&self) -> Vec<(String, &dyn Module)>

Get named children modules

pub fn named_parameters(&self) -> HashMap<String, Parameter>

Get named parameters

pub fn to_device(&mut self, device: DeviceType) -> Result<()>

Move to device

pub fn register_parameter(&mut self, name: String, param: Parameter)

Register a parameter

pub fn register_buffer(&mut self, name: String, tensor: Tensor)

Register a buffer

pub fn register_module(&mut self, name: String, module: Box<dyn Module>)

Register a submodule

pub fn all_parameter_tensors(&self) -> Vec<Arc<RwLock<Tensor>>>

Get all parameters including submodules (legacy method)

pub fn get_all_named_parameters(&self) -> HashMap<String, Parameter>

Get all parameters with module hierarchical names

pub fn all_named_parameters(&self) -> HashMap<String, Arc<RwLock<Tensor>>>

Get all named parameters including submodules

pub fn register_hook( &mut self, hook_type: HookType, callback: HookCallback, ) -> HookHandle

Register a hook for this module

pub fn remove_hook(&mut self, hook_type: HookType, handle: HookHandle) -> bool

Remove a hook by handle

pub fn execute_hooks( &self, hook_type: HookType, module: &dyn Module, input: &Tensor, output: Option<&Tensor>, ) -> Result<()>

Execute hooks of a specific type

pub fn has_hooks(&self, hook_type: HookType) -> bool

Check if any hooks are registered

pub fn hook_count(&self, hook_type: HookType) -> usize

Get hook count for a specific type

pub fn clear_hooks(&mut self, hook_type: HookType)

Clear all hooks of a specific type

pub fn clear_all_hooks(&mut self)

Clear all hooks

Trait Implementations

impl Debug for ModuleBase

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ModuleBase

fn default() -> Self

Returns the “default value” for a type.