diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java | 4 | ||||
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Slice.java | 4 |
2 files changed, 8 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java index fa3d70a4ddf..2a2551c9a58 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Generate.java @@ -71,6 +71,10 @@ public class Generate<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAM throw new IllegalArgumentException("A generated tensor can only have indexed bound dimensions"); } + public ScalarFunction<NAMETYPE> getBoundGenerator() { + return boundGenerator; + } + @Override public List<TensorFunction<NAMETYPE>> arguments() { return Collections.emptyList(); } diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Slice.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Slice.java index bccd66acd31..a0a3552eb92 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Slice.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Slice.java @@ -53,6 +53,10 @@ public class Slice<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMETY return new Slice<>(arguments.get(0), subspaceAddress); } + public List<DimensionValue<NAMETYPE>> getSubspaceAddress() { + return subspaceAddress; + } + @Override public PrimitiveTensorFunction<NAMETYPE> toPrimitive() { return this; } |