torch.Tensor.masked_scatter_

Tensor.masked_scatter_(mask, source)

Копирует элементы из source в тензор self в позициях, где mask равно True. Элементы из source копируются в self начиная с позиции 0 в source и далее по порядку, по одному для каждого вхождения mask, равного True. Форма mask должна быть совместима для трансляции с формой базового тензора. source должен содержать не менее элементов, чем количество единиц в mask.

Параметры:

• mask (BoolTensor) – булева маска
• source (Tensor) – тензор, из которого выполняется копирование

Примечание

Операция mask выполняется над тензором self, а не над переданным тензором source.

Пример

>>> self = torch.tensor([[0, 0, 0, 0, 0], [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]])