remove tranpose in set_buffer!

Author: hmyuuu

src/Algorithm/AD.jl

@@ -47,7 +47,7 @@ function update!(opt::StateOpt, alg::AbstractAD, obj, dynamics, output)
     (; max_episode) = alg
     f_ini, f_comp = objective(obj, dynamics)
     set_f!(output, f_ini)
-    set_buffer!(output, transpose(dynamics.data.ψ0))
+    set_buffer!(output,dynamics.data.ψ0)
     set_io!(output, f_ini)
     show(opt, output, obj)
     for ei in 1:(max_episode-1)
@@ -56,7 +56,7 @@ function update!(opt::StateOpt, alg::AbstractAD, obj, dynamics, output)
         dynamics.data.ψ0 = dynamics.data.ψ0/norm(dynamics.data.ψ0)
         f_out, f_now = objective(obj, dynamics)
         set_f!(output, f_out)
-        set_buffer!(output, transpose(dynamics.data.ψ0))
+        set_buffer!(output, dynamics.data.ψ0)
         set_io!(output, f_out, ei)
         show(output, obj)
     end
@@ -191,7 +191,7 @@ function update!(opt::StateControlOpt, alg::AbstractAD, obj, dynamics, output)
     f_ini, f_comp = objective(obj, dynamics)
 
     set_f!(output, f_ini)
-    set_buffer!(output, transpose(dynamics.data.ψ0), dynamics.data.ctrl)
+    set_buffer!(output, dynamics.data.ψ0, dynamics.data.ctrl)
     set_io!(output, f_noctrl, f_ini)
     show(opt, output, obj)
 
@@ -204,7 +204,7 @@ function update!(opt::StateControlOpt, alg::AbstractAD, obj, dynamics, output)
         f_out, f_now = objective(obj, dynamics)
 
         set_f!(output, f_out)
-        set_buffer!(output, transpose(dynamics.data.ψ0), dynamics.data.ctrl)
+        set_buffer!(output, dynamics.data.ψ0, dynamics.data.ctrl)
         set_io!(output, f_out, ei)
         show(output, obj)
     end

src/Algorithm/RI.jl

@@ -5,7 +5,7 @@ function update!(opt::StateOpt, alg::RI, obj, dynamics, output)
     f = QFIM(rho, drho)
 
     set_f!(output, f[1,1])
-    set_buffer!(output, transpose(dynamics.data.ψ0))
+    set_buffer!(output, dynamics.data.ψ0)
     set_io!(output, f[1,1])
     show(opt, output, obj)
 
@@ -24,7 +24,7 @@ function update!(opt::StateOpt, alg::RI, obj, dynamics, output)
         dynamics.data.ψ0 = psi0
 
         set_f!(output, f[1,1])
-        set_buffer!(output, transpose(dynamics.data.ψ0))
+        set_buffer!(output, dynamics.data.ψ0)
         set_io!(output, f[1,1], ei)
         show(output, obj)
     end