torch.Tensor.where

Tensor.where(condition, y) → Tensor

self.where(condition, y) 等价于 torch.where(condition, self, y)。请参阅 torch.where()