torch.can_cast
torch.can_cast(from, to) → bool-
Determines if a type conversion is allowed under PyTorch casting rules described in the type promotion documentation.
- Parameters
-
- from (dtype) – The original
torch.dtype. - to (dtype) – The target
torch.dtype.
- from (dtype) – The original
Example:
>>> torch.can_cast(torch.double, torch.float) True >>> torch.can_cast(torch.float, torch.int) False
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.can_cast.html