Diffusion Data Augmentation Tensorflow