diff --git a/src/mlpack/methods/ann/layer/concat_impl.hpp b/src/mlpack/methods/ann/layer/concat_impl.hpp index 208bdbfd0c..0b94e10f27 100644 --- a/src/mlpack/methods/ann/layer/concat_impl.hpp +++ b/src/mlpack/methods/ann/layer/concat_impl.hpp @@ -239,7 +239,7 @@ void ConcatType::Backward( delta.reshape(delta.n_elem / gy.n_cols, gy.n_cols); this->network[index]->Backward( - index == 0 ? input : this->layerOutputs[index-1], + input, this->layerOutputs[index], delta, g);