Merge pull request #3471 from mrdaybird/fix_jacobian

Fix JacobianTest and CustomJacobianTest (Resolves #3470)
mlpack · Apr 28, 2023 · a7cc156 · a7cc156
2 parents 403099f + bdd6893
commit a7cc156
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mlpack/tests/ann/ann_test_tools.hpp b/src/mlpack/tests/ann/ann_test_tools.hpp
@@ -74,7 +74,7 @@ double JacobianTest(ModuleType& module,
     derivTemp(i) = 1;
 
     arma::mat delta(input.n_rows, input.n_cols);
-    module.Backward(input, deriv, delta);
+    module.Backward(output, deriv, delta);
 
     jacobianB.col(i) = delta;
   }
@@ -125,7 +125,7 @@ double CustomJacobianTest(ModuleType& module,
     deriv(i) = 1;
 
     arma::mat delta(input.n_rows, input.n_cols);
-    module.Backward(input, deriv, delta);
+    module.Backward(output, deriv, delta);
 
     jacobianB.col(i) = delta;
   }