SelectScatter

Embeds values into the tensor at the given index.

Abstract Signature:

SelectScatter(input: Tensor, src: Tensor, dim: int, index: int)

PyTorchJAX (Core)Keras

PyTorch

API: torch.select_scatter

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (scatter_indexer)

Keras

API: keras.ops.scatter_update

Strategy: Direct Mapping

Official Docs ↗