tfa.image.angles_to_projective_transforms

Returns projective transform(s) for the given angle(s).

angles A scalar angle to rotate all images by, or (for batches of images) a vector with an angle to rotate each image in the batch. The rank must be statically known (the shape is not TensorShape(None).
image_height Height of the image(s) to be transformed.
image_width Width of the image(s) to be transformed.

A tensor of shape (num_images, 8). Projective transforms which can be given to transform op.