torch.can_cast#

torch.can_cast(from_,to)→bool#

Determines if a type conversion is allowed under PyTorch casting rulesdescribed in the type promotiondocumentation.

Parameters:

• from_ (dtype) – The originaltorch.dtype.

• to (dtype) – The targettorch.dtype.

Example:

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