GridSampler

Performs grid sampling for 2D or 3D inputs.

Abstract Signature:

GridSampler(input: Tensor, grid: Tensor, interpolation_mode: int = 0, padding_mode: int = 0, align_corners: bool = False)

PyTorchJAX (Core)TensorFlowApple MLX

PyTorch

API: torch.grid_sampler

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.scipy.ndimage.map_coordinates

Strategy: Plugin (grid_sample_adapter)

TensorFlow

API: —

Strategy: Custom / Partial

Apple MLX

API: —

Strategy: Custom / Partial