I've always thought that was much more logical, the current system in the UK just doesn't make sense at it's core => if you're an employer you can get saddled with an employee who you have to pay whilst they don't work for a period, who can then quit, but you're not allowed/expected to try to avoid this by not hiring women. Or taking it into account in promotions/salary.
If you really believe both of: 1: women should be paid to have children at their current salary. 2: women should not be discriminated against in any way. Then the only way to reconcile it is to have the government pay. And no, equal male leave wouldn't solve the problem because they don't take it. And no, mandatory male leave isn't a good solution. For a start it's unbelievably authoritarian to force someone to leave their work because they had a child. For a second you still have exactly the same discrimination problem it's just directed at potential parents instead of women. That would remove it as a feminist issue so it wouldn't get press but it would still be a problem.