summaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java
diff options
context:
space:
mode:
authorJon Bratseth <jonbratseth@yahoo.com>2017-01-10 20:16:39 +0100
committerGitHub <noreply@github.com>2017-01-10 20:16:39 +0100
commit2676b9c9987193f6c48677e30357123c4164eccc (patch)
treeaf9c84f76951392adc50c8dd4a552775d61cee36 /vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java
parent697fa08ac031476eee46f7b9cc8b902617ad2371 (diff)
Revert "Revert "Add (disabled) dense tensor binary format""
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;