Random Affine Transformation Tensorflow