fixed test case
This commit is contained in:
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user