move nn.functional to burn-tensor

Author: crutcher

crates/burn-core/src/nn/mod.rs

@@ -1,3 +1,5 @@
+pub use burn_tensor::nn::*;
+
 /// Attention module
 pub mod attention;
 
@@ -19,9 +21,6 @@ pub mod transformer;
 /// Interpolate module
 pub mod interpolate;
 
-/// Functional module
-pub mod functional;
-
 mod dropout;
 mod embedding;
 mod gelu;

crates/burn-tensor/src/tensor/mod.rs

@@ -32,6 +32,9 @@ pub mod loss;
 /// The burn module.
 pub mod module;
 
+/// The nn module.
+pub mod nn;
+
 /// Operations on tensors module.
 pub mod ops;
 

crates/burn-core/src/nn/functional/linear.rs renamed to crates/burn-tensor/src/tensor/nn/functional/linear.rs

@@ -0,0 +1,2 @@
+/// The functional module.
+pub mod functional;