torch.Tensor.where

Tensor.where(condition, y) → Tensor

self.where(condition, y) is equivalent to torch.where(condition, self, y). See torch.where()