GridSampler3d

Performs 3D grid sampling.

Abstract Signature:

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

PyTorchJAX (Core)TensorFlow

PyTorch

API: torch.grid_sampler_3d

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

TensorFlow

API: —

Strategy: Custom / Partial