Merge pull request #19 from aarontrowbridge/rollout-feature

Rollout feature
kestrelquantum · May 21, 2023 · 79cce44 · 79cce44
2 parents 5947ed2 + 466d01d
commit 79cce44
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/rollouts.jl b/src/rollouts.jl
@@ -45,6 +45,13 @@ function rollout(
     return Ψ̃
 end
 
+function rollout(
+    ψ̃₁s::AbstractVector{AbstractVector}, args...; kwargs...
+)
+    return vcat([rollout(ψ̃₁, args...; kwargs...) for ψ̃₁ ∈ ψ̃₁s]...)
+end
+
+
 function unitary_rollout(
     Ũ⃗₁::AbstractVector{<:Real},
     controls::AbstractMatrix{Float64},