实现 VAE 的重建与潜变量 KL 训练损失。