tf.keras.preprocessing.image.apply_affine_transform