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