pub fn diagonal_ensemble_density( eigenvalues: &[f64], eigenvectors: &[Vec<Complex64>], initial_density: &[Complex64], ) -> Result<DiagonalEnsemble>