I was very moved by Louis Theroux's Mothers on the Edge. It really brought back memories of how much I struggled after DC2's birth and really, my mental health was so crushed by psychosis and suicidal thoughts, I'm not sure how I avoided a mother and baby unit. My thoughts and loving wishes for the best future are with you and your baby, OP. 
Do you feel more at peace in the unit, like, you have room to breathe and just 'be', OP?
Is your partner and family supportive now and was he/your family supportive prior to your admission?