pub trait QuantumModule: Send + Sync {
// Required methods
fn forward(&mut self, input: &SciRS2Array) -> Result<SciRS2Array>;
fn parameters(&self) -> Vec<Parameter>;
fn train(&mut self, mode: bool);
fn training(&self) -> bool;
fn zero_grad(&mut self);
fn name(&self) -> &str;
}
Expand description
Base trait for all quantum ML modules
Required Methods§
Sourcefn forward(&mut self, input: &SciRS2Array) -> Result<SciRS2Array>
fn forward(&mut self, input: &SciRS2Array) -> Result<SciRS2Array>
Forward pass
Sourcefn parameters(&self) -> Vec<Parameter>
fn parameters(&self) -> Vec<Parameter>
Get all parameters