torch (version 0.8.1)

torch_can_cast: Can_cast

Description

Can_cast

Usage

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())
}