fixed test case

This commit is contained in:
Abhinav Anand
2021-05-27 22:48:07 +05:30
parent 5c1600c112
commit d8d6de550b
+1
View File
@@ -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);