Multiply batch dimension expansion tensor instead of adding it

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 83eb25d9dc2..1e6645df792 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(0));
+            ExpressionNode generatedExpression = new ConstantNode(new DoubleValue(1.0));
             Generate generatedFunction = new Generate(expandDimensionsType,
                                                       new GeneratorLambdaFunctionNode(expandDimensionsType,
                                                                                       generatedExpression)
                                                               .asLongListToDoubleOperator());
-            Join expand = new Join(TensorFunctionNode.wrapArgument(node), generatedFunction, ScalarFunctions.add());
+            Join expand = new Join(TensorFunctionNode.wrapArgument(node), generatedFunction, ScalarFunctions.multiply());
             return new TensorFunctionNode(expand);
         }
         return node;