summaryrefslogtreecommitdiffstats
path: root/config-model
diff options
context:
space:
mode:
authorHarald Musum <musum@oath.com>2018-03-22 18:33:17 +0100
committerGitHub <noreply@github.com>2018-03-22 18:33:17 +0100
commit441d2fd3a0f985200f688a6c387963b15ee13104 (patch)
treeae481e148686aae0cee6ff62b04774070942f255 /config-model
parent2b59453da99dccbf5380cf68efaea3918c01b437 (diff)
Revert "Multiply batch dimension expansion tensor instead of adding it"
Diffstat (limited to 'config-model')
-rw-r--r--config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/TensorFlowFeatureConverter.java4
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;