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 | 8 |
1 files changed, 8 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 e829f4c909b..13ddf3c2e20 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java @@ -129,6 +129,14 @@ public class TensorType { return dimensions.equals(((TensorType)other).dimensions); } + /** Returns whether the given type has the same dimension names as this */ + public boolean mathematicallyEquals(TensorType other) { + if (dimensions().size() != other.dimensions().size()) return false; + for (int i = 0; i < dimensions().size(); i++) + if (!dimensions().get(i).name().equals(other.dimensions().get(i).name())) return false; + return true; + } + @Override public int hashCode() { return dimensions.hashCode(); |