From be5b4e5e016a23afe7d1856dee500710c04084e2 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Wed, 20 Oct 2021 17:46:03 -0400
Subject: [PATCH] Fix shape of input to reflect the number of input maps.

---
 src/mlpack/tests/ann_layer_test.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp
index 84356b15b3..cf6f760784 100644
--- a/src/mlpack/tests/ann_layer_test.cpp
+++ b/src/mlpack/tests/ann_layer_test.cpp
@@ -4420,7 +4420,7 @@ TEST_CASE("ConvolutionLayerTestCase", "[ANNLayerTest]")
         << 32 << 13 << 42 << arma::endr;
 
   Convolution layer(4, 1, 1, 1, 1, 0, 0);
-  layer.InputDimensions() = std::vector<size_t>({ 4, 1 });
+  layer.InputDimensions() = std::vector<size_t>({ 4, 1, 2 });
   layer.ComputeOutputDimensions();
   arma::mat layerWeights(layer.WeightSize(), 1);
   layer.SetWeights(layerWeights.memptr());