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

Args
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.

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