From 1dab4e23a6cf5c92e22bdbe92d7ce170b08124d4 Mon Sep 17 00:00:00 2001
From: Adam Kropp <30504149+akropp@users.noreply.github.com>
Date: Wed, 1 Nov 2023 17:41:33 -0400
Subject: [PATCH] Update src/mlpack/methods/ann/layer/concat_impl.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
---
 src/mlpack/methods/ann/layer/concat_impl.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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<MatType>::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);