Implement normalize! for Product ansatz

bsc-quantic · Jan 22, 2024 · 284d68f · 284d68f
1 parent 65cb2c6
commit 284d68f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/Quantum/Product.jl b/src/Quantum/Product.jl
@@ -47,3 +47,15 @@ function LinearAlgebra.norm(ψ::Product{State}, p::Real = 2)
         mapreduce(Base.Fix2(^, p), +, parent(tensor))
     end^(1 // p)
 end
+
+function LinearAlgebra.normalize!(ψ::Product{State}, p::Real = 2; insert::Union{Nothing,Int} = nothing)
+    norm = LinearAlgebra.norm(ψ, p)
+
+    n = length(tensors(ψ))
+    norm ^= 1 / n
+    for tensor in tensors(ψ)
+        tensor ./= norm
+    end
+
+    ψ
+end