diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java index cf6e737bf27..aab4a4e1297 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java @@ -83,6 +83,16 @@ public class MixedTensor implements Tensor { } @Override + public Double getAsDouble(TensorAddress address) { + var block = index.blockOf(address); + int denseOffset = index.denseOffsetOf(address); + if (block == null || denseOffset < 0 || denseOffset >= block.cells.length) { + return null; + } + return block.cells[denseOffset]; + } + + @Override public boolean has(TensorAddress address) { var block = index.blockOf(address); int denseOffset = index.denseOffsetOf(address); |