tfa.image.translations_to_projective_transforms

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

tfa.image.translations_to_projective_transforms(
    translations: tfa.types.TensorLike,
    name: Optional[str] = None
) -> tf.Tensor

translations A 2-element list representing [dx, dy] or a matrix of 2-element lists representing [dx, dy] to translate for each image (for a batch of images). The rank must be statically known (the shape is not TensorShape(None)).
name The name of the op.

A tensor of shape (num_images, 8) projective transforms which can be given to tfa.image.transform.