diff options
author | Harald Musum <musum@oath.com> | 2018-03-22 18:33:17 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-03-22 18:33:17 +0100 |
commit | 441d2fd3a0f985200f688a6c387963b15ee13104 (patch) | |
tree | ae481e148686aae0cee6ff62b04774070942f255 /config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java | |
parent | 2b59453da99dccbf5380cf68efaea3918c01b437 (diff) |
Revert "Multiply batch dimension expansion tensor instead of adding it"
Diffstat (limited to 'config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java index 1e6645df792..83eb25d9dc2 100644 --- a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java +++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java @@ -383,12 +383,12 @@ public class TensorFlowFeatureConverter extends ExpressionTransformer<RankProfil } TensorType expandDimensionsType = typeBuilder.build(); if (expandDimensionsType.dimensions().size() > 0) { - ExpressionNode generatedExpression = new ConstantNode(new DoubleValue(1.0)); + ExpressionNode generatedExpression = new ConstantNode(new DoubleValue(0)); Generate generatedFunction = new Generate(expandDimensionsType, new GeneratorLambdaFunctionNode(expandDimensionsType, generatedExpression) .asLongListToDoubleOperator()); - Join expand = new Join(TensorFunctionNode.wrapArgument(node), generatedFunction, ScalarFunctions.multiply()); + Join expand = new Join(TensorFunctionNode.wrapArgument(node), generatedFunction, ScalarFunctions.add()); return new TensorFunctionNode(expand); } return node; |