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

This commit is contained in:
Omar Shrit
2023-04-12 04:47:23 +02:00
committed by GitHub
parent d58d9c7493
commit d8b26b92df
@@ -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);