diff --git a/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/recurrent/GravesBidirectionalLSTMTest.java b/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/recurrent/GravesBidirectionalLSTMTest.java index b95c13d9fb53..b928b95dc8dc 100644 --- a/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/recurrent/GravesBidirectionalLSTMTest.java +++ b/deeplearning4j-core/src/test/java/org/deeplearning4j/nn/layers/recurrent/GravesBidirectionalLSTMTest.java @@ -359,7 +359,7 @@ public void testSimpleForwardsAndBackwardsActivation() { final INDArray activation3Reverse = activation3.dup(); reverseColumnsInPlace(activation3Reverse); - assertArrayEquals(activation3Reverse.data().asFloat(),activation1.data().asFloat(),1e-5f); + assertEquals(activation3Reverse,activation1); assertArrayEquals(activation3Reverse.shape(),activation1.shape()); //test backprop now