MaskedScatter

Copies elements from source into self tensor at positions where the mask is True.

Abstract Signature:

MaskedScatter(input: Tensor, mask: Tensor, source: Tensor)

PyTorchJAX (Core)

PyTorch

API: torch.masked_scatter

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (scatter_indexer)