diff options
author | Lester Solbakken <lesters@oath.com> | 2021-06-21 12:21:19 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-06-21 12:21:19 +0200 |
commit | 626870d56dcd8b4b199a643f29880b948641da8e (patch) | |
tree | e94997b49bc9dcab66209260b16b712120ddb7eb /vespajlib | |
parent | 642cdd49f80c8d3f1f590c1c59ec4e6f1d699bf3 (diff) |
Add bit ranking function in Java
Diffstat (limited to 'vespajlib')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/Tensor.java | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java index 3133752bc49..ab475e25387 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java @@ -240,6 +240,7 @@ public interface Tensor { default Tensor equal(Tensor argument) { return join(argument, (a, b) -> ( a == b ? 1.0 : 0.0)); } default Tensor notEqual(Tensor argument) { return join(argument, (a, b) -> ( a != b ? 1.0 : 0.0)); } default Tensor approxEqual(Tensor argument) { return join(argument, (a, b) -> ( approxEquals(a,b) ? 1.0 : 0.0)); } + default Tensor bit(Tensor argument) { return join(argument, (a,b) -> ((int)b < 8 && (int)b >= 0 && ((int)a & (1 << (int)b)) != 0) ? 1.0 : 0.0); } default Tensor avg() { return avg(Collections.emptyList()); } default Tensor avg(String dimension) { return avg(Collections.singletonList(dimension)); } |