fix #5

QuantumBFS · Sep 23, 2019 · 821e76d · 821e76d
1 parent 2545ea2
commit 821e76d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Diff.jl b/src/Diff.jl
@@ -198,7 +198,7 @@ function backward_params!(st, block::Diff{<:DiffBlock}, collector)
     in, outδ = st
     Σ = generator(content(block))
     g = dropdims(sum(conj.(statevec(in |> Σ)) .* statevec(outδ), dims=1), dims=1)
-    pushfirst!(collector, -g |> imag)
+    pushfirst!(collector, -g[1] |> imag)
     in |> Σ
     nothing
 end