tfa.image.angles_to_projective_transforms

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

tfa.image.angles_to_projective_transforms(
    angles: tfa.types.TensorLike,
    image_height: tfa.types.TensorLike,
    image_width: tfa.types.TensorLike,
    name: Optional[str] = None
) -> tf.Tensor

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.