diff options
Diffstat (limited to 'model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java index adb54474812..3211a44fa68 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Join.java @@ -82,6 +82,13 @@ public class Join extends IntermediateOperation { bReducedFunction = new Reduce(b.function().get(), Reduce.Aggregator.sum, bDimensionsToReduce); } + // retain order of inputs + if (a == inputs.get(1)) { + TensorFunction temp = bReducedFunction; + bReducedFunction = aReducedFunction; + aReducedFunction = temp; + } + return new com.yahoo.tensor.functions.Join(aReducedFunction, bReducedFunction, operator); } |