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)

PyTorch

API: torch.slice_scatter
Strategy: Direct Mapping
Official Docs ↗

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)