PyTorch
torch / torch
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
.
Example:
>>> torch.can_cast(torch.double, torch.float) True >>> torch.can_cast(torch.float, torch.int) False
此页内容是否对您有帮助
感谢反馈!