Trait MLPLayer 

pub trait MLPLayer: Send + Sync {
    // Required method
    fn forward(&self, hidden_states: &Tensor) -> Result<Tensor>;
}

MLP trait for expert networks in MoE layers

Required Methods

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

Forward pass through the MLP

Dyn Compatibility

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors