torch.Tensor.where

Tensor.where(condition, y) → Tensor

self.where(condition, y) 等價於 torch.where(condition, self, y)。請參閱 torch.where()