diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/TensorType.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/TensorType.java | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java index acba9eafd71..bb7a976af9b 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java @@ -24,9 +24,15 @@ import java.util.stream.Collectors; */ public class TensorType { + public enum ValueType { DOUBLE, FLOAT}; + /** The empty tensor type - which is the same as a double */ public static final TensorType empty = new TensorType(Collections.emptyList()); + private final ValueType valueType = ValueType.DOUBLE; + + public final ValueType valueType() { return valueType; } + /** Sorted list of the dimensions of this */ private final ImmutableList<Dimension> dimensions; |