diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java new file mode 100644 index 00000000000..cb0615693ae --- /dev/null +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java @@ -0,0 +1,29 @@ +package com.yahoo.tensor.functions; + +import java.util.function.DoubleBinaryOperator; +import java.util.function.DoubleUnaryOperator; + +/** + * The join tensor function. + * + * @author bratseth + */ +public class Map extends PrimitiveTensorFunction { + + private final TensorFunction argument; + private final DoubleUnaryOperator mapper; + + public Map(TensorFunction argument, DoubleUnaryOperator mapper) { + this.argument = argument; + this.mapper = mapper; + } + + public TensorFunction argument() { return argument; } + public DoubleUnaryOperator mapper() { return mapper; } + + @Override + public PrimitiveTensorFunction toPrimitive() { + return new Map(argument.toPrimitive(), mapper); + } + +} |