torch.Tensor.masked_scatter_

Tensor.masked_scatter_(mask, source)

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

Параметры:

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

Примечание

Функция mask работает с тензором self, а не с заданным тензором source.