Fixup contraction with N-body terms

QuantumKitHub · Jul 2, 2024 · fb4ea7d · fb4ea7d
1 parent 9e26775
commit fb4ea7d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/operators/localoperator.jl b/src/operators/localoperator.jl
@@ -134,7 +134,9 @@ const PEPS_ENVBONDDIM = :(χ^2)
         )
     end
 
-    operator = tensorexpr(:O, (:O_out_1, :O_out_2), (:O_in_1, :O_in_2))
+    operator = tensorexpr(
+        :O, ntuple(i -> Symbol(:O_out_, i), N), ntuple(i -> Symbol(:O_in_, i), N)
+    )
 
     bra = map(Iterators.product(1:gridsize[1], 1:gridsize[2])) do (i, j)
         inds_id = findfirst(==(CartesianIndex(xmin + i - 1, ymin + j - 1)), cartesian_inds)