fix(initializer): init shape (f_in, f_out)

drewxs · Jul 23, 2023 · f47e866 · f47e866
1 parent 812e3fa
commit f47e866
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/initializer.rs b/src/initializer.rs
@@ -33,10 +33,10 @@ impl Initializer {
     /// Initializes a 2D tensor with random values based on the specified standard deviation.
     fn initialize_data(f_in: usize, f_out: usize, std_dev: f64) -> Tensor2D {
         let mut rng = rand::thread_rng();
-        let mut data = Vec::with_capacity(f_out);
-        for _ in 0..f_out {
-            let mut row = Vec::with_capacity(f_in);
-            for _ in 0..f_in {
+        let mut data = Vec::with_capacity(f_in);
+        for _ in 0..f_in {
+            let mut row = Vec::with_capacity(f_out);
+            for _ in 0..f_out {
                 row.push(rng.gen::<f64>() * std_dev);
             }
             data.push(row);