SliceUpdate

Update a sub-array of the input array.

Abstract Signature:

SliceUpdate(x: Tensor, update: Tensor, start_indices: Union[List[int], Tensor], axes: Optional[List[int]])

PyTorchJAX (Core)Apple MLX

PyTorch

API: —

Strategy: Plugin (slice_update_adapter)

JAX (Core)

API: jax.lax.dynamic_update_slice

Strategy: Plugin (slice_update_adapter)

Apple MLX

API: mlx.core.slice_update

Strategy: Direct Mapping

Official Docs ↗