Struct QuantumLSTM 

pub struct QuantumLSTM { /* private fields */ }

LSTM layer

Implementations

impl QuantumLSTM

pub fn new(input_size: usize, hidden_size: usize) -> Self

Create new LSTM

pub fn num_layers(self, num_layers: usize) -> Self

Set number of layers

pub fn bidirectional(self, bidirectional: bool) -> Self

Set bidirectional

pub fn dropout(self, dropout: f64) -> Self

Set dropout

pub fn batch_first(self, batch_first: bool) -> Self

Set batch first

pub fn forward_with_state( &mut self, input: &SciRS2Array, initial_state: Option<LSTMState>, ) -> Result<(SciRS2Array, LSTMState)>

Forward pass with optional initial state

Trait Implementations

impl QuantumModule for QuantumLSTM

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

Forward pass

fn parameters(&self) -> Vec<Parameter>

Get all parameters

fn train(&mut self, mode: bool)

Set training mode

fn training(&self) -> bool

Check if module is in training mode

fn zero_grad(&mut self)

Zero gradients of all parameters

fn name(&self) -> &str

Module name for debugging