From d8b26b92dfcaa1f93c40f77d21cb2a2df83bc9be Mon Sep 17 00:00:00 2001 From: Omar Shrit Date: Wed, 12 Apr 2023 04:47:23 +0200 Subject: [PATCH] Fix the casting from default double to any possible type (#3467) --- src/mlpack/methods/ann/init_rules/network_init.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/mlpack/methods/ann/init_rules/network_init.hpp b/src/mlpack/methods/ann/init_rules/network_init.hpp index 4c923d6dc6..d0810b7840 100644 --- 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 tmp = arma::mat(parameters.memptr() + offset, + arma::Mat tmp = arma::Mat(parameters.memptr() + offset, weight, 1, false, false); initializeRule.Initialize(tmp, tmp.n_elem, 1);