diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/IndexedFloatTensor.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/IndexedFloatTensor.java | 9 |
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 |