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)`` .. raw:: html

PyTorch

API: torch.affine_grid_generator
Strategy: Direct Mapping

JAX (Core)

API:
Strategy: Custom / Partial

Keras

API:
Strategy: Custom / Partial

TensorFlow

API: tf.raw_ops.AffineGrid
Strategy: Direct Mapping

Apple MLX

API:
Strategy: Custom / Partial