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.

Parameters

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

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

Example:

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