pytorch
432724b3 - Fix torch.where to accept only tensors with same dtypes(CPU) (#29078)

Commit
6 years ago
Fix torch.where to accept only tensors with same dtypes(CPU) (#29078) * Make zeros argument of torch.where same dtype as other argument * Added check for torch.where on CPU that both arguments have same dtype * Changes based on PR comments * Fix flake8 * Fixed test for CUDA * Changes basen on PR comments * Changes based on PR review
Author
Committer
Parents
Loading