burn-nn 0.22.0

Neural network building blocks for the Burn deep learning framework
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! LSTM module.
//!
//! This module implements LSTM cells and variants.
//!
//! # References
//!
//! - [Long Short-Term Memory](https://en.wikipedia.org/wiki/Long_short-term_memory)
//! - [A Simple Neural Network and Deep Learning Algorithm for Prediction of Stock Prices](https://www.researchgate.net/publication/221444045_A_Simple_Neural_Network_and_Deep_Learning_Algorithm_for_Prediction_of_Stock_Prices)
//! - [Understanding LSTM Networks](https://colah.github.io/posts/2015-08-Understanding-LSTMs/)
mod bilstm_module;
#[cfg(test)]
mod checkpoint_tests;
mod lstm_module;
mod lstm_state;

#[doc(inline)]
pub use bilstm_module::*;
#[doc(inline)]
pub use lstm_module::*;
#[doc(inline)]
pub use lstm_state::*;