Police are investigating an incident in which officers grabbed a baby from a mother during her arrest.