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 (dpython:type) – The original
torch.dtype. -
to (dpython:type) – The target
torch.dtype.
-
from (dpython:type) – The original
Example:
>>> torch.can_cast(torch.double, torch.float) True >>> torch.can_cast(torch.float, torch.int) False
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.can_cast.html