Trait Module 

pub trait Module {
    // Required methods
    fn forward(&self, input: &Tensor) -> Result<Tensor>;
    fn parameters(&self) -> Vec<Rc<RefCell<Tensor>>>;
    fn box_clone(&self) -> Box<dyn Module>;

    // Provided methods
    fn train(&mut self) { ... }
    fn eval(&mut self) { ... }
}

Trait para módulos de red neuronal

Similar a torch.nn.Module, define la interfaz para:

• forward() - propagación hacia adelante
• parameters() - obtener todos los parámetros entrenables Trait para módulos de red neuronal

Similar a torch.nn.Module, define la interfaz para:

• forward() - propagación hacia adelante
• parameters() - obtener todos los parámetros entrenables

Required Methods

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

Forward pass del módulo

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

Devuelve referencias a todos los parámetros entrenables

fn box_clone(&self) -> Box<dyn Module>

Clone the module (helper for Box)

Provided Methods

fn train(&mut self)

Pone el módulo en modo entrenamiento

fn eval(&mut self)

Pone el módulo en modo evaluación

Trait Implementations

impl Clone for Box<dyn Module>

fn clone(&self) -> Box<dyn Module>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

Implementors

impl Module for BatchNorm1d

impl Module for Conv1d

impl Module for Dropout

impl Module for Flatten

impl Module for Linear

impl Module for ReLU

impl Module for Sequential

impl Module for Sigmoid

impl Module for Softmax