From d8d6de550bbbe7647a29e3186fb9a8c8bbbbabe4 Mon Sep 17 00:00:00 2001
From: Abhinav Anand <abhinav.anand2807@gmail.com>
Date: Thu, 27 May 2021 22:48:07 +0530
Subject: [PATCH] fixed test case

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

diff --git a/src/mlpack/tests/ann_layer_test.cpp b/src/mlpack/tests/ann_layer_test.cpp
index 2bc3eff890..63c82f3b4a 100644
--- a/src/mlpack/tests/ann_layer_test.cpp
+++ b/src/mlpack/tests/ann_layer_test.cpp
@@ -4693,6 +4693,7 @@ TEST_CASE("ChannelShuffleLayerTest", "[ANNLayerTest]")
   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);