Não só a vagina, e sim a musculatura pélvica. Essa contração involuntária faz parte do

orgasmo e ocorre para facilitar a fecundação, como já comentaram.