Function partial_trace 

pub fn partial_trace(
    state: &[Complex64],
    subsystem_dimension: usize,
    environment_dimension: usize,
) -> Result<Vec<Complex64>>

Reduced density matrix of the first factor of a pure bipartite state.