edit

kestrelquantum · Jun 20, 2023 · 2926add · 2926add
1 parent 1cb67a8
commit 2926add
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/quantum_utils.jl b/src/quantum_utils.jl
@@ -89,8 +89,13 @@ function qubit_system_state(ket::String)
     return ψ
 end
 
-function lift(U, q, n; l=2)
-    Is = Matrix{Number}[I(l) for i in 1:n]
+function lift(
+    U::AbstractMatrix{<:Number},
+    q::Int,
+    n::Int;
+    l::Int=2
+)
+    Is = Matrix{Number}[I(l) for _ = 1:n]
     Is[q] = U
     return foldr(kron, Is)
 end