add rectified linear

kkalass · May 1, 2015 · 5c82e93 · 5c82e93
1 parent 6c99430
commit 5c82e93
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/factory/DefaultOpFactory.java b/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/factory/DefaultOpFactory.java
@@ -129,6 +129,8 @@ public Accumulation createAccum(String name, INDArray x, INDArray y) {
     @Override
     public TransformOp createTransform(String name, INDArray x, INDArray y) {
         switch (name) {
+            case "relu":
+                return new RectifedLinear(x,0);
             case "abs":
                 return new Abs(x, y);
             case "acos":