AffineGridGenerator

Generates a 2D or 3D flow field (sampling grid), given a batch of affine matrices.

Abstract Signature:

AffineGridGenerator(theta: Tensor, size: List[int], align_corners: bool = False)

PyTorchJAX (Core)KerasTensorFlowApple MLX

PyTorch

API: torch.affine_grid_generator

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

Keras

API: —

Strategy: Custom / Partial

TensorFlow

API: tf.raw_ops.AffineGrid

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial