Update src/mlpack/methods/ann/layer/concat_impl.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
2023-11-01 17:41:33 -04:00
parent c7d94a6ff3
commit 1dab4e23a6
1 changed files with 1 additions and 1 deletions
@@ -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);