Make tensor generate inspectable

1 files changed, 4 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/TensorFunction.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/TensorFunction.java
index b4c5dedbf4e..5c0d0a99441 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/TensorFunction.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/TensorFunction.java
@@ -9,6 +9,7 @@ import com.yahoo.tensor.evaluation.Name;
 import com.yahoo.tensor.evaluation.TypeContext;
 
 import java.util.List;
+import java.util.Optional;
 
 /**
  * A representation of a tensor function which is able to be translated to a set of primitive
@@ -61,6 +62,9 @@ public abstract class TensorFunction<NAMETYPE extends Name> {
      */
     public abstract String toString(ToStringContext context);
 
+    /** Returns this as a scalar function, or empty if it cannot be represented as a scalar function */
+    public Optional<ScalarFunction<NAMETYPE>> asScalarFunction() { return Optional.empty(); }
+
     @Override
     public String toString() { return toString(ToStringContext.empty()); }