Tensor function language skeleton

1 files changed, 30 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Join.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Join.java
new file mode 100644
index 00000000000..421ea631ee3
--- /dev/null
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Join.java
@@ -0,0 +1,30 @@
+package com.yahoo.tensor.functions;
+
+import java.util.function.DoubleBinaryOperator;
+
+/**
+ * The join tensor function. 
+ * 
+ * @author bratseth
+ */
+public class Join extends PrimitiveTensorFunction {
+    
+    private final TensorFunction argumentA, argumentB;
+    private final DoubleBinaryOperator combinator;
+
+    public Join(TensorFunction argumentA, TensorFunction argumentB, DoubleBinaryOperator combinator) {
+        this.argumentA = argumentA;
+        this.argumentB = argumentB;
+        this.combinator = combinator;
+    }
+
+    public TensorFunction argumentA() { return argumentA; }
+    public TensorFunction argumentB() { return argumentB; }
+    public DoubleBinaryOperator combinator() { return combinator; }
+    
+    @Override
+    public PrimitiveTensorFunction toPrimitive() {
+        return new Join(argumentA.toPrimitive(), argumentB.toPrimitive(), combinator);
+    }
+
+}