torch.can_cast

torch.can_cast(from, to) → bool

确定在 PyTorch 类型提升规则下是否允许类型转换，详见类型提升的文档。

Parameters

• 从 (数据类型) – 原始的 torch.dtype。

• 到 (数据类型) – 目标 torch.dtype.

示例：

>>> torch.can_cast(torch.double, torch.float)
真
>>> torch.can_cast(torch.float, torch.int)
假