//! A small neural-network library based on Torch.
//!
//! This library tries to stay as close as possible to the original
//! Python and C++ implementations.
mod var_store;
pub use var_store::Path;
pub use var_store::VarStore;

mod module;
pub use module::{Module, ModuleT};

mod linear;
pub use linear::Linear;

mod conv2d;
pub use conv2d::{Conv2D, Conv2DConfig};

mod conv_transpose2d;
pub use conv_transpose2d::{ConvTranspose2D, ConvTranspose2DConfig};

mod batch_norm;
pub use batch_norm::BatchNorm2D;

mod rnn;
pub use rnn::{GRU, LSTM, RNN};

mod func;
pub use func::{Func, FuncT};

mod sequential;
pub use sequential::Sequential;
pub use sequential::SequentialT;

mod c_optimizer;
pub mod optimizer;
pub use optimizer::Optimizer;

pub struct Id();
impl ModuleT for Id {
    fn forward_t(&self, xs: &crate::Tensor, _train: bool) -> crate::Tensor {
        xs.shallow_clone()
    }
}
