Believers, will babies be/think like babies in heaven/after life?

I’m not really asking this question towards people believing in organized religion like Christianity, Buddhism, or Muslim but more towards those who just have their own idea of faith.

Do you believe babies that die will think the same way once they’re in the afterlife/heaven? I have a hard time thinking of this one.

I prefer those who believe in an afterlife to answer which makes an interesting discussion.