Struct LSTM 

pub struct LSTM { /* private fields */ }

LSTM layer (Keras-compatible)

Implementations

impl LSTM

pub fn new(units: usize) -> Self

Create new LSTM layer

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

Set return sequences

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

Set return state

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

Set go backwards

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

Set dropout

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

Set recurrent dropout

pub fn name(self, name: &str) -> Self

Set layer name

Trait Implementations

impl KerasLayer for LSTM

fn call(&self, input: &ArrayD<f64>) -> Result<ArrayD<f64>>

Forward pass through the layer

fn build(&mut self, input_shape: &[usize]) -> Result<()>

Build the layer (called during model compilation)

fn compute_output_shape(&self, input_shape: &[usize]) -> Vec<usize>

Compute output shape given input shape

fn count_params(&self) -> usize

Get number of parameters

fn get_weights(&self) -> Vec<ArrayD<f64>>

Get trainable parameters

fn set_weights(&mut self, weights: Vec<ArrayD<f64>>) -> Result<()>

Set trainable parameters

fn built(&self) -> bool

Check if layer is built

fn name(&self) -> &str

Get layer name