GridSampler2d

Performs 2D grid sampling.

Abstract Signature:

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

PyTorchJAX (Core)TensorFlow

PyTorch

API: torch.grid_sampler_2d

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.scipy.ndimage.map_coordinates

Strategy: Plugin (grid_sample_adapter)

TensorFlow

API: —

Strategy: Custom / Partial