Add check in TensorNetwork constructor for inconsistent dimensions

bsc-quantic · Jul 12, 2023 · 71506a5 · 71506a5
1 parent 6f21d05
commit 71506a5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/TensorNetwork.jl b/src/TensorNetwork.jl
@@ -38,6 +38,14 @@ struct TensorNetwork{A<:Ansatz,M<:NamedTuple}
             mergewith(vcat, dict, Dict([index => [i] for index in labels(tensor)]))
         end
 
+        # Check for inconsistent dimensions
+        for (index, idxs) in indices
+            sizes = [size(tensors[i], index) for i in idxs]
+            if !all(x -> x == sizes[1], sizes)
+                throw(DimensionMismatch("Different sizes specified for index $index"))
+            end
+        end
+
         M = Tenet.metadata(A)
         metadata = M((; metadata...))