torch.can_cast¶
- torch.can_cast(from_, to) bool ¶
确定在类型提升文档中描述的 PyTorch 类型转换规则下是否允许类型转换。
- 参数
from_ (dtype) – 原始的
torch.dtype
。to (dtype) – 目标
torch.dtype
。
示例
>>> torch.can_cast(torch.double, torch.float) True >>> torch.can_cast(torch.float, torch.int) False