From 86123a1e751d516e9b9c2d8a46af406df7e586be Mon Sep 17 00:00:00 2001
From: abh2k <41710346+abh2k@users.noreply.github.com>
Date: Fri, 28 May 2021 00:04:25 +0530
Subject: [PATCH]  Fix test case

---
 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 63c82f3b4a..a8d7c4605a 100644
--- a/src/mlpack/tests/ann_layer_test.cpp
+++ b/src/mlpack/tests/ann_layer_test.cpp
@@ -4689,11 +4689,11 @@ TEST_CASE("ChannelShuffleLayerTest", "[ANNLayerTest]")
                   << 6  << 22 << arma::endr
                   << 7  << 23 << arma::endr
                   << 8  << 24 << arma::endr;
+  outputExpected1.reshape(24, 1);
   // Check the Forward pass of the layer.
   module1.Forward(input1, output1);
   CheckMatrices(output1, outputExpected1);
 
-  outputExpected1.reshape(24, 1);
   // Check the Backward pass of the layer.
   module1.Backward(output1, output1, outputBackward1);
   CheckMatrices(input1, outputBackward1);