pub fn partial_trace( matrix: &Array2<Complex64>, keep_qubits: &[usize], total_qubits: usize, ) -> QuantRS2Result<Array2<Complex64>>
Compute the partial trace of a matrix