Slice

Extract a sub-array from the input array using start indices and sizes.

Abstract Signature:

Slice(x: Tensor, start_indices: Union[List[int], Tensor], slice_sizes: Union[List[int], Tensor], axes: Optional[List[int]])

PyTorchJAX (Core)KerasApple MLX

PyTorch

API: —

Strategy: Plugin (slice_adapter)

JAX (Core)

API: jax.lax.dynamic_slice

Strategy: Plugin (slice_adapter)

Keras

API: keras.ops.slice

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.core.slice

Strategy: Direct Mapping

Official Docs ↗