Hi, Could you please explain why the encoders output is z_dim * 2 and not just z_dim. Thanks for the very clean repo.