a bit more elegant

USCqserver · Dec 9, 2023 · ea673d4 · ea673d4
1 parent 237043d
commit ea673d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/math_util.jl b/src/math_util.jl
@@ -228,7 +228,7 @@ function partial_trace(ρ::AbstractMatrix, sys_dim::AbstractVector{<:Integer}, d
 	re_dim[dim_2_trace] .= 1
 	tr_dim = copy(sys_dim)
 	tr_dim[dim_2_keep] .= 1
-	res = zeros(typeof(first(ρ)), re_dim..., re_dim...)
+	res = zeros(eltype(ρ), re_dim..., re_dim...)
 	for I in CartesianIndices(size(res))
 		for k in CartesianIndices((tr_dim...,))
 			delta = CartesianIndex(k, k)