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/1.13/generated/torch.can_cast.html
