diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2024-02-01 12:27:18 +0100 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2024-02-01 12:27:18 +0100 |
commit | 8441c97b9f8c56dd937eb2aa50a4cd7d7bb9aa62 (patch) | |
tree | 5854b6dfd843adda0f73af53595593b514b39f3f /vespajlib/src/main | |
parent | 216ccc5cf7746a1039ca364727f2138b75572f71 (diff) |
Reverse the problem to match the interface of TensorAddress.partialCopy
Diffstat (limited to 'vespajlib/src/main')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java index ed4154464fc..05db61f5395 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java @@ -89,7 +89,7 @@ public class Rename<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMET for (int i = 0; i < tensor.type().dimensions().size(); i++) { String dimensionName = tensor.type().dimensions().get(i).name(); String newDimensionName = fromToMap.getOrDefault(dimensionName, dimensionName); - toIndexes[i] = renamedType.indexOfDimension(newDimensionName).get(); + toIndexes[renamedType.indexOfDimension(newDimensionName).get()] = i; } // avoid building a new tensor if dimensions can simply be renamed @@ -100,7 +100,7 @@ public class Rename<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMET Tensor.Builder builder = Tensor.Builder.of(renamedType); for (Iterator<Tensor.Cell> i = tensor.cellIterator(); i.hasNext(); ) { Map.Entry<TensorAddress, Double> cell = i.next(); - TensorAddress renamedAddress = rename(cell.getKey(), toIndexes); + TensorAddress renamedAddress = cell.getKey().partialCopy(toIndexes); builder.cell(renamedAddress, cell.getValue()); } return builder.build(); @@ -118,10 +118,6 @@ public class Rename<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMET return true; } - private static TensorAddress rename(TensorAddress address, int[] toIndexes) { - return address.partialCopy(toIndexes); - } - private String toVectorString(List<String> elements) { if (elements.size() == 1) return elements.get(0); |