diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions/Join.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Join.java | 30 |
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); + } + +} |