torch.Tensor.where
- Tensor.where(condition, y) → Tensor
-
self.where(condition, y)等同于torch.where(condition, self, y)。详情参见torch.where()
本页目录
•
pytorch-doc-2.5中文文档
self.where(condition, y) 等同于 torch.where(condition, self, y)。详情参见 torch.where()