Scatter

Writes values into a tensor at indices specified by index.

Abstract Signature:

Scatter(input: Tensor, dim: int, index: Tensor, src: Tensor, reduce: str)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLX

PyTorch

API: torch.scatter

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.ops.index_update

Strategy: Plugin (scatter_indexer)

NumPy

API: numpy.put_along_axis

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.ops.scatter

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.tensor_scatter_nd_update

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial