summaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java
diff options
context:
space:
mode:
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java')
-rw-r--r--vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java8
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;