Allow for building index on the outside.

1 files changed, 4 insertions, 1 deletions

diff --git a/eval/src/vespa/eval/tensor/dense/direct_dense_tensor_builder.h b/eval/src/vespa/eval/tensor/dense/direct_dense_tensor_builder.h
index ac53ddf5e95..935b5c20373 100644
--- a/eval/src/vespa/eval/tensor/dense/direct_dense_tensor_builder.h
+++ b/eval/src/vespa/eval/tensor/dense/direct_dense_tensor_builder.h
@@ -31,7 +31,10 @@ public:
     DirectDenseTensorBuilder(const eval::ValueType &type_in);
     ~DirectDenseTensorBuilder();
     void insertCell(const Address &address, double cellValue) {
-        _cells[calculateCellAddress(address, _type)] = cellValue;
+        insertCell(calculateCellAddress(address, _type), cellValue);
+    }
+    void insertCell(size_t index, double cellValue) {
+        _cells[index] = cellValue;
     }
     Tensor::UP build();
 };