Why do babies need to learn to submerge at such a young age?

I'm not sure of the reasoning behind it... However, my personal opinion is that if the baby is scared, going underwater for a second or two and coming back up unharmed may help to show them that there isn't anything there to fear. Also, it might help get their body temperature better adjusted to the water as the water may seem colder if only part of you is wet.