aboutsummaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
diff options
context:
space:
mode:
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/Tensor.java')
-rw-r--r--vespajlib/src/main/java/com/yahoo/tensor/Tensor.java7
1 files changed, 7 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 59d5ee72372..483ccd330e0 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
@@ -86,6 +86,13 @@ public interface Tensor {
return valueIterator().next();
}
+ /**
+ * Returns this tensor with the given type if types are compatible
+ *
+ * @throws IllegalArgumentException if types are not compatible
+ */
+ Tensor withType(TensorType type);
+
// ----------------- Primitive tensor functions
default Tensor map(DoubleUnaryOperator mapper) {