diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java b/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java index daa85cc51e4..7570a357452 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java @@ -29,6 +29,14 @@ public final class DimensionSizes { /** Returns the number of dimensions this provides the size of */ public int dimensions() { return sizes.length; } + /** Returns the product of the sizes of this */ + public int totalSize() { + int productSize = 1; + for (int dimensionSize : sizes ) + productSize *= dimensionSize; + return productSize; + } + @Override public boolean equals(Object o) { if (o == this) return true; |