Struct QuantumGRU 

pub struct QuantumGRU { /* private fields */ }

GRU layer

Implementations

impl QuantumGRU

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

Create new GRU

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

Set number of layers

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

Set bidirectional

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

Set batch first

pub fn forward_with_hidden( &mut self, input: &SciRS2Array, initial_hidden: Option<SciRS2Array>, ) -> Result<(SciRS2Array, SciRS2Array)>

Forward pass with optional initial hidden state

Trait Implementations

impl QuantumModule for QuantumGRU

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