(See e.g. https://en.wikipedia.org/wiki/Sex_differences_in_psychology#... - Childhood play, Intelligence, Personality traits, Empathy)
Science very heavily favors breastfeeding above the substitute. While pumping breast milk is a choice, it can be difficult in practice and some science points to it still being different for both the mother and the child in terms of hormonal differences and of course the physican touch/emotional wellbeing.
How do society treat men who stay?
How do society treat women who leave?
How do society treat men who stay?
Society in this case mean social status, divorce rates, size of social networks, number of children, and other measurable factors. Answer all four and we likely have a clear answer why men don't leave at the same rate as women after they have a child.