API

 torch / torch


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

Example:

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

此页内容是否对您有帮助