Diffusion Model Tensorflow Tutorial Mnist