Fix softmax for RnnOutputLayer

jungleTail · Mar 20, 2016 · b6d8947 · b6d8947
1 parent 266c498
commit b6d8947
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/recurrent/RnnOutputLayer.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/recurrent/RnnOutputLayer.java
@@ -144,9 +144,7 @@ public INDArray output(boolean training){
         INDArray preOutput2d = preOutput2d(training);
 
         if(conf.getLayer().getActivationFunction().equals("softmax")) {
-            SoftMax softMax = new SoftMax(preOutput2d);
-            softMax.exec(1);
-            INDArray out2d = softMax.z();
+            INDArray out2d = Nd4j.getExecutioner().execAndReturn(new SoftMax(preOutput2d));
             if(maskArray != null){
                 out2d.muliColumnVector(maskArray);
             }