SliceScatter ============ Embeds values into the tensor at a given dimension slice. **Abstract Signature:** ``SliceScatter(input: Tensor, src: Tensor, dim: int = 0, start: int, end: int, step: int = 1)`` .. raw:: html

PyTorch

API: torch.slice_scatter
Strategy: Direct Mapping

JAX (Core)

API: jax.lax.dynamic_update_slice
Strategy: Plugin (slice_scatter_plugin)

NumPy

API:
Strategy: Custom / Partial

Keras

API: keras.ops.slice_update
Strategy: Plugin (slice_scatter_plugin)

TensorFlow

API: tf.tensor_scatter_nd_update
Strategy: Plugin (slice_scatter_plugin)

Apple MLX

API:
Strategy: Custom / Partial

Flax NNX

API: jax.lax.dynamic_update_slice
Strategy: Plugin (slice_scatter_plugin)

PaxML / Praxis

API: jax.lax.dynamic_update_slice
Strategy: Plugin (slice_scatter_plugin)