pub fn reshape_process_matrix( process_matrix: &Array4<Complex64>, target_shape: (usize, usize), ) -> DeviceResult<Array2<Complex64>>
Reshape process matrix for different representations