Augmented KRnet for density estimation and approximation