diff options
author | Jon Bratseth <bratseth@yahoo-inc.com> | 2016-11-16 13:43:01 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@yahoo-inc.com> | 2016-11-16 13:43:01 +0100 |
commit | 9d8296953e573fc23fe4e346219d4155e6f4e81c (patch) | |
tree | 62a770a165002b5e096ce75c03aad0c48358cd54 /vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java | |
parent | 4ad513c134bf980431d14f1c2c1d4775086047ec (diff) |
More functions
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java index d571875796b..d3fc707b65d 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/L1Normalize.java @@ -6,14 +6,19 @@ package com.yahoo.tensor.functions; public class L1Normalize extends CompositeTensorFunction { private final TensorFunction argument; + private final String dimension; - public L1Normalize(TensorFunction argument) { + public L1Normalize(TensorFunction argument, String dimension) { this.argument = argument; + this.dimension = dimension; } @Override public PrimitiveTensorFunction toPrimitive() { - return new Join(argument.toPrimitive(), new Reduce(argument.toPrimitive(), Reduce.Aggregator.avg, "dimension"), ScalarFunctions.multiply()); + TensorFunction primitiveArgument = argument.toPrimitive(); + return new Join(primitiveArgument, + new Reduce(primitiveArgument, Reduce.Aggregator.avg, dimension), + ScalarFunctions.multiply()); } @Override |