Allow passing your own vector without copy to the IndexedTensor.

1 files changed, 8 insertions, 1 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/IndexedFloatTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/IndexedFloatTensor.java
index 8f8c24c8421..56cb22da7a5 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/IndexedFloatTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/IndexedFloatTensor.java
@@ -43,8 +43,15 @@ class IndexedFloatTensor extends IndexedTensor {
         private float[] values;
 
         BoundFloatBuilder(TensorType type, DimensionSizes sizes) {
+            this(type, sizes, new float[(int)sizes.totalSize()]);
+        }
+        BoundFloatBuilder(TensorType type, DimensionSizes sizes, float [] values) {
             super(type, sizes);
-            values = new float[(int)sizes.totalSize()];
+            if (sizes.totalSize() != values.length) {
+                throw new IllegalArgumentException("Invalid size("  + values.length + ") of supplied value vector." +
+                                                   " Type specifies that size should be " + sizes.totalSize());
+            }
+            this.values = values;
         }
 
         @Override