More functions

1 files changed, 32 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/L2Normalize.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/L2Normalize.java
new file mode 100644
index 00000000000..eb632ee679a
--- /dev/null
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/L2Normalize.java
@@ -0,0 +1,32 @@
+package com.yahoo.tensor.functions;
+
+/**
+ * @author bratseth
+ */
+public class L2Normalize extends CompositeTensorFunction {
+
+    private final TensorFunction argument;
+    private final String dimension;
+    
+    public L2Normalize(TensorFunction argument, String dimension) {
+        this.argument = argument;
+        this.dimension = dimension;
+    }
+    
+    @Override
+    public PrimitiveTensorFunction toPrimitive() {
+        TensorFunction primitiveArgument = argument.toPrimitive();
+        return new Join(primitiveArgument, 
+                        new Map(new Reduce(new Map(primitiveArgument, ScalarFunctions.square()), 
+                                           Reduce.Aggregator.sum,
+                                           dimension), 
+                                ScalarFunctions.square()), 
+                        ScalarFunctions.divide());
+    }
+    
+    @Override
+    public String toString() {
+        return "l2_normalize(" + argument + ")";
+    }
+
+}