Adding basic RNN and GRU implementation

coreylowman · Jun 25, 2023 · 83053b1 · 83053b1
1 parent 780b347
commit 83053b1
Show file tree

Hide file tree

Showing 2 changed files with 711 additions and 0 deletions.
diff --git a/src/nn/mod.rs b/src/nn/mod.rs
@@ -190,6 +190,8 @@ mod batchnorm2d;
 mod bias2d;
 #[cfg(feature = "nightly")]
 mod conv;
+#[cfg(feature = "nightly")]
+mod rec;
 mod convtrans;
 mod dropout;
 mod ema;
@@ -284,6 +286,10 @@ pub mod builders {
     pub use super::dropout::{Dropout, DropoutOneIn};
     pub use super::embedding::builder::Embedding;
     #[cfg(feature = "nightly")]
+    pub use super::rec::builder::RNN;
+    #[cfg(feature = "nightly")]
+    pub use super::rec::builder::GRU;
+    #[cfg(feature = "nightly")]
     pub use super::flatten::Flatten2D;
     pub use super::generalized_residual::GeneralizedResidual;
     pub use super::layer_norm::builder::LayerNorm1D;