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