2d Diffusion Model