Fix VarAutoEncoder reparameterize returning mu+std at inference (#8413)

Fix VarAutoEncoder reparameterize returning mu+std at inference (#8413) #340