FLAT MANIFOLD VAES