Retrieve bipartite reduced density matrix

Hi Brian, Aditya, here are some notes that hopefully answer this question. Brian, sorry I never got back to you last summer when you asked the follow-up question.