pub mod init;
pub use init::{f_init, init, Init};
mod var_store;
pub use var_store::{Path, VarStore, Variables};
mod module;
pub use module::{Module, ModuleT};
mod linear;
pub use linear::*;
mod conv;
pub use conv::*;
mod conv_transpose;
pub use conv_transpose::*;
mod batch_norm;
pub use batch_norm::*;
mod group_norm;
pub use group_norm::*;
mod layer_norm;
pub use layer_norm::*;
mod sparse;
pub use sparse::*;
mod rnn;
pub use rnn::*;
mod func;
pub use func::*;
mod sequential;
pub use sequential::*;
mod optimizer;
pub use optimizer::{
adam, adamw, rms_prop, sgd, Adam, AdamW, Optimizer, OptimizerConfig, RmsProp, Sgd,
};
#[derive(Debug)]
pub struct Id();
impl ModuleT for Id {
fn forward_t(&self, xs: &crate::Tensor, _train: bool) -> crate::Tensor {
xs.shallow_clone()
}
}
impl Module for crate::CModule {
fn forward(&self, xs: &crate::Tensor) -> crate::Tensor {
self.forward_ts(&[xs]).unwrap()
}
}
impl ModuleT for crate::TrainableCModule {
fn forward_t(&self, xs: &crate::Tensor, _train: bool) -> crate::Tensor {
self.inner.forward_ts(&[xs]).unwrap()
}
}