diff options
author | Jon Bratseth <jonbratseth@yahoo.com> | 2017-01-10 20:16:39 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-01-10 20:16:39 +0100 |
commit | 2676b9c9987193f6c48677e30357123c4164eccc (patch) | |
tree | af9c84f76951392adc50c8dd4a552775d61cee36 /vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java | |
parent | 697fa08ac031476eee46f7b9cc8b902617ad2371 (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.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; |