fix reshape in diagonal contract

euro-hpc-pl · Apr 3, 2024 · 089163c · 089163c
1 parent ee95376
commit 089163c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/contractions/diagonal.jl b/src/contractions/diagonal.jl
@@ -31,5 +31,5 @@ function contract_matrix_tensor3(C::DiagonalTensor{R}, B::Tensor{R,3}) where {R<
     B = contract_matrix_tensor3(C.e2, B)
     # @cast B[l, r, (q1, q2)] := B[l, (r, q1), q2] (q1 ∈ 1:size(C.e1, 1))
     q1 = size(C.e1, 1)
-    B = reshape(B, size(B, 1), size(B, 2) ÷ q2, size(B, 3) * q1)
+    B = reshape(B, size(B, 1), size(B, 2) ÷ q1, size(B, 3) * q1)
 end