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.

参数

• from_ (dtype) – The original torch.dtype.

• to (dtype) – The target torch.dtype.

示例

>>> torch.can_cast(torch.double, torch.float)
True
>>> torch.can_cast(torch.float, torch.int)
False