Function partial_trace 

pub fn partial_trace(
    state: &TwoQubitState,
    trace_out: usize,
) -> DensityMatrix2x2

Partial trace: trace out one qubit to get the density matrix of the other.