torch.Tensor.masked_scatter — PyTorch 2.7 documentation (original) (raw)

Tensor.masked_scatter(mask, tensor) → Tensor¶

Out-of-place version of torch.Tensor.masked_scatter_()

Example

self = torch.tensor([0, 0, 0, 0, 0]) mask = torch.tensor([[0, 0, 0, 1, 1], [1, 1, 0, 1, 1]], dtype=torch.bool) source = torch.tensor([[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]) self.masked_scatter(mask, source) tensor([[0, 0, 0, 0, 1], [2, 3, 0, 4, 5]])