torch_can_cast function

Can_cast

torch_can_cast(from_, to)

Arguments

• from_: (dtype) The original torch_dtype.
• to: (dtype) The target torch_dtype.

can_cast(from, to) -> bool

Determines if a type conversion is allowed under PyTorch casting rules described in the type promotion documentation .

Examples

if (torch_is_installed()) {

torch_can_cast(torch_double(), torch_float())
torch_can_cast(torch_float(), torch_int())
}