MaskedScatterΒΆ

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

Abstract Signature:

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

PyTorch

API: torch.masked_scatter
Strategy: Direct Mapping

JAX (Core)

API: β€”
Strategy: Plugin (scatter_indexer)