Fix the casting from default double to any possible type (#3467)

mlpack · Apr 12, 2023 · d8b26b9 · d8b26b9
1 parent d58d9c7
commit d8b26b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mlpack/methods/ann/init_rules/network_init.hpp b/src/mlpack/methods/ann/init_rules/network_init.hpp
@@ -71,7 +71,7 @@ class NetworkInitialization
         // Initialize the layer with the specified parameter/weight
         // initialization rule.
         const size_t weight = network[i]->WeightSize();
-        arma::Mat<eT> tmp = arma::mat(parameters.memptr() + offset,
+        arma::Mat<eT> tmp = arma::Mat<eT>(parameters.memptr() + offset,
             weight, 1, false, false);
         initializeRule.Initialize(tmp, tmp.n_elem, 1);