Refactor tensorflow import and add dropout test

1 files changed, 9 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/ScalarFunctions.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/ScalarFunctions.java
index 15d4dfc1d00..92c1c0307ec 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/ScalarFunctions.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/ScalarFunctions.java
@@ -33,6 +33,7 @@ public class ScalarFunctions {
     public static DoubleUnaryOperator acos() { return new Acos(); }
     public static DoubleUnaryOperator elu() { return new Elu(); }
     public static DoubleUnaryOperator exp() { return new Exp(); }
+    public static DoubleUnaryOperator floor() { return new Floor(); }
     public static DoubleUnaryOperator relu() { return new Relu(); }
     public static DoubleUnaryOperator rsqrt() { return new Rsqrt(); }
     public static DoubleUnaryOperator selu() { return new Selu(); }
@@ -126,6 +127,14 @@ public class ScalarFunctions {
         public String toString() { return "f(a)(exp(a))"; }
     }
 
+    public static class Floor implements DoubleUnaryOperator {
+        @Override
+        public double applyAsDouble(double operand) { return Math.floor(operand); }
+        @Override
+        public String toString() { return "f(a)(floor(a))"; }
+    }
+
+
     public static class Relu implements DoubleUnaryOperator {
         @Override
         public double applyAsDouble(double operand) { return Math.max(operand, 0); }